From f1f98af3c6bfaf49e3e45b7bb5e08f13aed69db7 Mon Sep 17 00:00:00 2001
From: BRIGGAN2 <briggan2@unl.edu>
Date: Mon, 9 Nov 2020 02:33:06 +0000
Subject: [PATCH] fixed line 42
---
cnn.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/cnn.py b/cnn.py
index 6592e5b..debd08a 100755
--- a/cnn.py
+++ b/cnn.py
@@ -39,7 +39,7 @@ def accuracy(labels, pred, k=1):
loss: average cross_entropy loss over batch (scalar)
'''
labels = np.argmax(labels, axis=-1)
- pred = np.argmax(labels, axis=-1)
+ pred = np.argmax(pred, axis=-1)
return np.mean(pred == labels)
--
GitLab