vulnerability_2_1

Running

leavoigt commited on Jan 4

Commit

73f34f0

•

1 Parent(s): 55c1b89

Update utils/target_classifier.py

Files changed (1) hide show

utils/target_classifier.py CHANGED Viewed

@@ -114,7 +114,7 @@ def target_classification(haystack_doc:pd.DataFrame,
         predictions = classifier_model(list(haystack_doc.text))
         # Get labels for predictions
-        pred_labels = getlabels(predictions)
         # Save labels
         haystack_doc['Target Label'] = pred_labels

         predictions = classifier_model(list(haystack_doc.text))
         # Get labels for predictions
+        pred_labels = gettarget_labels(predictions)
         # Save labels
         haystack_doc['Target Label'] = pred_labels