andre/PyTorch-Tutorial
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update accuracy function

Browse Source
This commit is contained in:
morvanzhou
2018-06-29 10:09:52 +08:00
parent 79380fb90e
commit f2ebd8a37b
3 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tutorial-contents/401_CNN.py
View File

@ -115,8 +115,8 @@ for epoch in range(EPOCH):
if step % 50 == 0:
test_output, last_layer = cnn(test_x)
pred_y = torch.max(test_output, 1)[1].data.squeeze()
accuracy = float(sum(pred_y == test_y)) / float(test_y.size(0))
pred_y = torch.max(test_output, 1)[1].data.squeeze().numpy()
accuracy = float((pred_y == test_y.data.numpy()).astype(int).sum()) / float(test_y.size(0))
print('Epoch: ', epoch, '| train loss: %.4f' % loss.data.numpy(), '| test accuracy: %.2f' % accuracy)
if HAS_SK:
# Visualization of trained flatten layer (T-SNE)