ml-digits-recognition 0.0.4

A simple digits recognition neural network

Convolutional Neural Network using MNIST dataset for Digit Recognition

Official repository: https://github.com/MartinBraquet/ml-digits-recognition.

Test online: https://martinbraquet.com/index.php/solo_page_digits_recognition.

Installation from PyPI

pip install ml-digits-recognition

Usage

from ml_digits_recognition import drawing
drawing.run()

Installation from Source

pip install -r requirements.txt

Documentation

Click here for a full description.

Visualization of the convolutional neural network:

nn_visualization.ipynb

Training

Train the model and save it as model.pt.

ml_digits_recognition_training.ipynb

Accuracy vs epochs.

Loss vs epochs.

Test

Test in Jupiter Notebook. The model can be loaded from the training above in model.pt or from the default precise model in model_precise.pt.

ml_digits_recognition_test.ipynb

Test in Python.

python src/ml_digits_recognition/drawing.py

Tools

Draw a digit and save it as a PNG file.

user_input_drawing.ipynb

Issues / Bug reports / Feature requests

Please open an issue.

Contributions

Contributions are welcome. Please check the outstanding issues and feel free to open a pull request.

Contributors