https://github.com/saranshmanu/learning-pytorch-cnn-mnist

This project help to learn the use Pytorch framework to create a simple Convolutional Neura Network and train the machine learning model on the classical MNIST dataset to recognise the handwritten digits.
https://github.com/saranshmanu/learning-pytorch-cnn-mnist

Last synced: about 2 months ago
JSON representation

This project help to learn the use Pytorch framework to create a simple Convolutional Neura Network and train the machine learning model on the classical MNIST dataset to recognise the handwritten digits.

• Host: GitHub
• URL: https://github.com/saranshmanu/learning-pytorch-cnn-mnist
• Owner: saranshmanu
• Created: 2019-05-06T20:16:29.000Z (about 6 years ago)
• Default Branch: master
• Last Pushed: 2019-05-17T11:16:10.000Z (about 6 years ago)
• Last Synced: 2025-02-01T07:19:02.762Z (4 months ago)
• Language: Jupyter Notebook
• Homepage:
• Size: 11 MB
• Stars: 0
• Watchers: 2
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# Learning-Pytorch-CNN-MNIST

This project help to learn the use Pytorch framework to create a simple Convolutional Neura Network and train the machine learning model on the classical MNIST dataset to recognise the handwritten digits.

PyTorch is an open-source machine learning library for Python, based on Torch, used for applications such as natural language processing. It is primarily developed by Facebook's artificial-intelligence research group, and Uber's "Pyro" Probabilistic programming language software is built on it.

## Results after training

### Evaluating the prediction model

![Screenshot](docs/result.png)

### Predicting the result with the handwritten image

![Screenshot](docs/prediction.png)

## Resources

https://adventuresinmachinelearning.com/convolutional-neural-networks-tutorial-in-pytorch/