test

app.py

@@ -13,7 +13,7 @@ def compute(data):
     print(data)
     # return metric.compute()
     result = {
-        "predictions": [list(map(int, pred.split(","))) for pred in data["predictions"]],
+        "predictions": [list(map(float, pred.split(","))) for pred in data["predictions"]],
         "references": data["references"].cast(pl.Int64).to_list()
     }
     print(result)
@@ -23,8 +23,8 @@ def compute(data):
 local_path = Path(sys.path[0])
 
 default_value = pl.DataFrame({
-    'predictions': ['1,2,3,4,5', '1,2,3,4,5', '1,2,3,4,5'],
-    'references': ['0', '1', '2']
+    'predictions': ['0.82,0.95,0.6,0.14,0.15,0.70', '0.67,0.31,0.01,0.60,0.44,0.51', '0.57,0.06,0.69,0.07,0.96,0.72'],
+    'references': ['1', '3', '4']
 })
 
 iface = gr.Interface(

top5_error_rate.py

@@ -66,6 +66,8 @@ class Top5ErrorRate(evaluate.Metric):
         # to numpy array
         outputs = np.array(predictions)
         labels = np.array(references)
+        print(outputs)
+        print(labels)
 
         # Top-1 ACC
         pred = outputs.argmax(axis=1)