update GHG

appStore/ghg.py

@@ -21,13 +21,6 @@ import plotly.express as px
 classifier_identifier = 'ghg'
 params  = get_classifier_params(classifier_identifier)
 
-# Labels dictionary ###
-_lab_dict = {
-            'NEGATIVE':'NO GHG TARGET',
-            'NA':'NOT APPLICABLE',
-            'TARGET':'GHG TARGET',
-            }
-
 
 def app():
     ### Main app code ###

utils/ghg_classifier.py

@@ -10,10 +10,9 @@ from transformers import pipeline
 
 # Labels dictionary ###
 _lab_dict = {
-                      'LABEL_0':'NEGATIVE',
-                      'LABEL_1':'NOT GHG',
-                      'LABEL_2':'GHG',
-                      'NA':'NA',
+                      'GHG':'GHG',
+                      'NOT_GHG':'NON GHG TRANSPORT TARGET',
+                      'NEGATIVE':'OTHERS',
                       }
         
 
@@ -85,7 +84,7 @@ def ghg_classification(haystack_doc:pd.DataFrame,
     results = classifier_model(list(temp.text))
     labels_= [(l[0]['label'],l[0]['score']) for l in results]
     temp['GHG Label'],temp['GHG Score'] = zip(*labels_)
-    
+    temp['GHG Label'] = temp['GHG Label'].apply(lambda x: _lab_dict[x])
     # merge back Target and non-Target dataframe
     df = pd.concat([df,temp])
     df['GHG Label'] = df['GHG Label'].apply(lambda i: _lab_dict[i])