-
A simple multilayer Perceptron (MLP) or simply one hidden layer neural network (NN) developed using PyTorch. It is trained and tested on MNIST dataset with average test accuracy around 95%.
-
A 2D Convolutional Neural Net (CNN) developed using pytorch. It is trained and tested on MNIST dataset. It reaches accuracy level over 99% after 10 epochs. The train loss, test loss and test accuracy with epoch number have been plotted, in Jupyter Notebook.
-
Notifications
You must be signed in to change notification settings - Fork 0
nkay28/Neural_networks
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
Neural Networks with PyTorch: MLP, CNN, LSTM.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published