This is a neural network that is trained on the MNIST hand written didgit data set. This neural network has been trained over 100 epochs and has an accuracy of 99%. You can provide a number and it will show you the drawing the net is trying to predict.
- requires pytorch and matplotlib
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
pip install matplotlib
git clone https://github.com/windyGarlic/NeuralNetwork_Identify_digits
cd NeuralNetwork_Identify_digits
python3 main.py