action policy split

utils/policyaction_classifier.py

@@ -86,13 +86,13 @@ def policyaction_classification(haystack_doc:pd.DataFrame,
     temp = truth_df.copy()
     truth_df = truth_df.astype(float) >= threshold
     truth_df = truth_df.astype(str)
-    truth_df.rename(columns = {'Action':'Action Label','Policies_Plans':'Policies_Plans Label'},
+    truth_df.rename(columns = {'Action':'Action Label','Policies & Plans':'Policies_Plans Label'},
                                  inplace =True)
     truth_df['Action Label'] = truth_df['Action Label'].apply(lambda x: 'Action' if 
                                                           True else 'NEGATIVE')
     truth_df['Policies_Plans Label'] = truth_df['Policies_Plans Label'].apply(lambda x: 
                                             'Policy and Plans' if True else 'NEGATIVE')
-    temp.rename(columns  = {'Action':'Action Score','Policies_Plans':'Policies_Plans Score'},
+    temp.rename(columns  = {'Action':'Action Score','Policies & Plans':'Policies_Plans Score'},
                           inplace=True)
     
     # categories = list(truth_df.columns)