labels is list, should be numpy array

uclnlp · Oct 25, 2017 · 6538e64 · 6538e64
1 parent 9121c63
commit 6538e64
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/jack/util/hooks.py b/jack/util/hooks.py
@@ -430,8 +430,9 @@ def apply_metrics(self, tensors: Mapping[TensorPort, np.ndarray]) -> Mapping[str
         labels = tensors[Ports.Target.target_index]
         predictions = tensors[Ports.Prediction.candidate_index]
 
-        acc_exact = np.sum(np.equal(labels, predictions))
-        acc_f1 = f1_score(np.array(labels), np.array(predictions), average='macro') * labels.shape[0]
+        labels_np = np.array(labels)
+        acc_exact = np.sum(np.equal(labels_np, predictions))
+        acc_f1 = f1_score(labels_np, predictions, average='macro') * labels_np.shape[0]
 
         return {"F1_macro": acc_f1, "Accuracy": acc_exact}