add conditionality

utils/conditional_classifier.py

@@ -8,12 +8,6 @@ from utils.preprocessing import processingpipeline
 import streamlit as st
 from transformers import pipeline
 
-# Labels dictionary ###
-_lab_dict = {
-            'NEGATIVE':'NO NETZERO TARGET',
-            'NET-ZERO':'NETZERO TARGET',
-            'TARGET_FREE':'OTHERS'
-            }
 
 @st.cache_resource
 def load_conditionalClassifier(config_file:str = None, classifier_name:str = None):
@@ -70,7 +64,7 @@ def conditional_classification(haystack_doc:pd.DataFrame,
     """
     logging.info("Working on Conditionality Identification")
     haystack_doc['Conditional Label'] = 'NA'
-    haystack_doc['Conditional Score'] = 'NA'
+    haystack_doc['Conditional Score'] = 0.0
     haystack_doc['cond_check'] = False
     haystack_doc['cond_check'] = haystack_doc.apply(lambda x: True if (
                 (x['Target Label'] == 'TARGET') | (x['Action Label'] == 'Action') |
@@ -91,6 +85,7 @@ def conditional_classification(haystack_doc:pd.DataFrame,
     # temp[' Label'] = temp['Netzero Label'].apply(lambda x: _lab_dict[x])
     # merging Target with Non Target dataframe
     df = pd.concat([df,temp])
+    df = df.drop(columns = ['cond_check'])
     df = df.reset_index(drop =True)
     df.index += 1