vulnerability_2_1

Sleeping

leavoigt commited on Sep 26, 2023

Commit

1970e55

•

1 Parent(s): 2df6f09

Update utils/vulnerability_classifier.py

Files changed (1) hide show

utils/vulnerability_classifier.py CHANGED Viewed

@@ -47,7 +47,7 @@ def getlabels(preds):
       if index_of_one != "NA":
         name  = label_dict[index_of_one]
       else:
-        name = "NA"
       # Append name to list
       predictions_names.append(name)

       if index_of_one != "NA":
         name  = label_dict[index_of_one]
       else:
+        name = "Other"
       # Append name to list
       predictions_names.append(name)