prop

IQTLabs · Oct 5, 2024 · 44bf0cb · 44bf0cb
1 parent e9b427d
commit 44bf0cb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/count_labels.py b/utils/count_labels.py
@@ -23,6 +23,6 @@
 
 prop = round(correct_labels / labels * 100, 2)
 print(correct_labels, labels, prop)
-if prop < prop_thresh:
+if prop < float(prop_thresh):
     print("predicted class is less than threshold")
     sys.exit(-1)