84
edits
Changes
→Implementation of a Neural Network
''' 8. Test the trained neural network to check whether the digit value shown in the image is in fact the number
''' the neural network determined it to be.
print(torch.argmax(net(X[0].view(-1, 784))[0]))