ydeng9 commited on
Commit
e48f9ee
·
verified ·
1 Parent(s): 0a8ca09

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -74,7 +74,7 @@ category_names = [
74
  prob_vector = probabilities[0].tolist() # shape: (12,)
75
 
76
  predicted_labels = []
77
- for cat_name, prob, label in zip(category_names, prob_vector):
78
  label = 1 if prob > threshold else 0
79
  predicted_labels.append(label)
80
 
 
74
  prob_vector = probabilities[0].tolist() # shape: (12,)
75
 
76
  predicted_labels = []
77
+ for cat_name, prob in zip(category_names, prob_vector):
78
  label = 1 if prob > threshold else 0
79
  predicted_labels.append(label)
80