Spaces:

TeresaK
/

cpv_test

Runtime error

leavoigt commited on Jan 5

Commit

6fef1b7

•

1 Parent(s): ab71d04

Update utils/target_classifier.py

Files changed (1) hide show

utils/target_classifier.py CHANGED Viewed

@@ -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.
     """
     # Get label names
     predictions_names=[]
     # loop through each prediction
-    for ele in preds:
       # see if there is a value 1 and retrieve index
       try:

     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_list:
       # see if there is a value 1 and retrieve index
       try: