Aye10032 commited on
Commit
53ccfce
·
1 Parent(s): 59d0cdb

update nptype

Browse files
Files changed (1) hide show
  1. 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)