Spaces:

GIZ
/

cpu_tracs

Sleeping

ppsingh commited on Jul 24, 2023

Commit

e86aae8

•

1 Parent(s): b1a66f3

add adaptation mitigation

Files changed (1) hide show

utils/adapmit_classifier.py CHANGED Viewed

@@ -69,7 +69,7 @@ def adapmit_classification(haystack_doc:pd.DataFrame,
                 False, axis=1)
     # we apply Netzero to only paragraphs which are classified as 'Target' related
     df1 = haystack_doc[haystack_doc['cond_check'] == True]
-    df1 = temp.reset_index(drop=True)
     df = haystack_doc[haystack_doc['cond_check'] == False]
     df = df.reset_index(drop=True)

                 False, axis=1)
     # we apply Netzero to only paragraphs which are classified as 'Target' related
     df1 = haystack_doc[haystack_doc['cond_check'] == True]
+    df1 = df1.reset_index(drop=True)
     df = haystack_doc[haystack_doc['cond_check'] == False]
     df = df.reset_index(drop=True)