Spaces:
Sleeping
Sleeping
update nptype
Browse files- top5_error_rate.py +1 -1
top5_error_rate.py
CHANGED
@@ -64,7 +64,7 @@ class Top5ErrorRate(evaluate.Metric):
|
|
64 |
**kwargs,
|
65 |
) -> Dict[str, Any]:
|
66 |
# to numpy array
|
67 |
-
outputs = np.array(predictions)
|
68 |
labels = np.array(references)
|
69 |
print(outputs)
|
70 |
print(labels)
|
|
|
64 |
**kwargs,
|
65 |
) -> Dict[str, Any]:
|
66 |
# to numpy array
|
67 |
+
outputs = np.array(predictions, dtype=np.float32)
|
68 |
labels = np.array(references)
|
69 |
print(outputs)
|
70 |
print(labels)
|