vulnerability_2_1

Sleeping

leavoigt commited on Jan 5

Commit

86c29fa

•

1 Parent(s): bb4307a

Update utils/target_classifier.py

Files changed (1) hide show

utils/target_classifier.py CHANGED Viewed

@@ -37,7 +37,7 @@ def get_target_labels(preds):
         index_of_one = ele.index(1)
       except ValueError:
         index_of_one = "NA"
       # Retrieve the name of the label (if no prediction made = NA)
       if index_of_one != "NA":
         name  = label_dict[index_of_one]

         index_of_one = ele.index(1)
       except ValueError:
         index_of_one = "NA"
+      st.write(index_of_one)
       # Retrieve the name of the label (if no prediction made = NA)
       if index_of_one != "NA":
         name  = label_dict[index_of_one]