accuracy function

yongqyu · Feb 23, 2019 · e801923 · e801923
1 parent 85f74ab
commit e801923
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/metric.py b/metric.py
@@ -0,0 +1,8 @@
+import torch
+from sklearn.metrics import accuracy_score
+
+def accuracy(preds, target):
+	preds = torch.max(preds, 1)[1].float()
+	acc = accuracy_score(preds.cpu().numpy(), target.cpu().numpy())
+
+	return acc