Update utils/target_classifier.py

utils/target_classifier.py

@@ -21,12 +21,15 @@ def get_target_labels(preds):
     Function that takes the numerical predictions as an input and returns a list of the labels.
     
     """
+
+    # Turn into list 
+    preds_list = list(preds)
     
     # Get label names
     predictions_names=[]
     
     # loop through each prediction
-    for ele in preds:
+    for ele in preds_list:
     
       # see if there is a value 1 and retrieve index
       try: