We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9370b18 commit af05b1bCopy full SHA for af05b1b
modelbuilders_bench/mb_utils.py
@@ -23,7 +23,7 @@ def get_accuracy(true_labels, prediction):
23
errors = 0
24
for i, true_label in enumerate(true_labels):
25
pred_label = 0
26
- if isinstance(prediction[i], (float, np.single, np.float)):
+ if isinstance(prediction[i], (np.float32, np.float64)):
27
pred_label = prediction[i] > 0.5
28
elif prediction[i].shape[0] == 1:
29
pred_label = prediction[i][0]
0 commit comments