ydeng9 commited on
Commit
a7419b9
·
verified ·
1 Parent(s): 494f6df

Update README.md

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