vulnerability

Sleeping

leavoigt commited on Sep 26, 2023

Commit

aaa7178

1 Parent(s): e000f68

Update utils/vulnerability_classifier.py

Files changed (1) hide show

utils/vulnerability_classifier.py CHANGED Viewed

@@ -75,7 +75,7 @@ def vulnerability_classification(haystack_doc:pd.DataFrame,
     the number of times it is covered/discussed/count_of_paragraphs.
     """
     logging.info("Working on vulnerability Identification")
-    haystack_doc['Sector Label'] = 'NA'
     # haystack_doc['PA_check'] = haystack_doc['Policy-Action Label'].apply(lambda x: True if len(x) != 0 else False)
     # df1 = haystack_doc[haystack_doc['PA_check'] == True]

     the number of times it is covered/discussed/count_of_paragraphs.
     """
     logging.info("Working on vulnerability Identification")
+    haystack_doc['Vulnerability Label'] = 'NA'
     # haystack_doc['PA_check'] = haystack_doc['Policy-Action Label'].apply(lambda x: True if len(x) != 0 else False)
     # df1 = haystack_doc[haystack_doc['PA_check'] == True]