Spaces:
Sleeping
Sleeping
change output type
Browse files- app.py +1 -1
- top5_error_rate.py +2 -2
app.py
CHANGED
@@ -21,7 +21,7 @@ local_path = Path(sys.path[0])
|
|
21 |
|
22 |
default_value = pl.DataFrame({
|
23 |
'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'],
|
24 |
-
'references': ['1', '3', '
|
25 |
})
|
26 |
|
27 |
iface = gr.Interface(
|
|
|
21 |
|
22 |
default_value = pl.DataFrame({
|
23 |
'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'],
|
24 |
+
'references': ['1', '3', '1']
|
25 |
})
|
26 |
|
27 |
iface = gr.Interface(
|
top5_error_rate.py
CHANGED
@@ -77,6 +77,6 @@ class Top5ErrorRate(evaluate.Metric):
|
|
77 |
top5_error_rate = 1 - correct.mean()
|
78 |
|
79 |
return {
|
80 |
-
"accuracy": acc,
|
81 |
-
"top5_error_rate": top5_error_rate
|
82 |
}
|
|
|
77 |
top5_error_rate = 1 - correct.mean()
|
78 |
|
79 |
return {
|
80 |
+
"accuracy": float(acc),
|
81 |
+
"top5_error_rate": float(top5_error_rate)
|
82 |
}
|