Update utils/target_classifier.py

utils/target_classifier.py

@@ -14,6 +14,38 @@ _lab_dict = {
             '1':'YES',
             }
 
+def get_target_labels(preds):
+
+    """
+    Function that takes the numerical predictions as an input and returns a list of the labels.
+    
+    """
+    
+    # Get label names
+    preds_list = preds.tolist()
+    
+    predictions_names=[]
+    
+    # loop through each prediction
+    for ele in preds_list:
+    
+      # see if there is a value 1 and retrieve index
+      try:
+        index_of_one = ele.index(1)
+      except ValueError:
+        index_of_one = "NA"
+    
+      # Retrieve the name of the label (if no prediction made = NA)
+      if index_of_one != "NA":
+        name  = label_dict[index_of_one]
+      else:
+        name = "Other"
+    
+      # Append name to list
+      predictions_names.append(name)
+    
+    return predictions_names
+
 @st.cache_resource
 def load_targetClassifier(config_file:str = None, classifier_name:str = None):
     """