Created
August 13, 2018 07:23
-
-
Save omarsar/b945599c88c14556a233bbf879456e52 to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| NN = Neural_Network() | |
| for i in range(1000): # trains the NN 1,000 times | |
| print ("#" + str(i) + " Loss: " + str(torch.mean((y - NN(X))**2).detach().item())) # mean sum squared loss | |
| NN.train(X, y) | |
| NN.saveWeights(NN) | |
| NN.predict() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment