1 metric update should be in the for loop. 2 sample_num is the shape's first element 3 preds cast should use round, astype("int32") method will cast all the elements(which is less than 1) to 0.