Machine Learning Playground

mlpg stands for Machine Learning Playground

It is a hands-on repository for self-learning of machine learning algorithms by solving real problems using pytorch and other ML libraries.

The MNIST database of handwritten digits, available from this page.

You can see the MNISTModel model in the models.py module, with an example of usage in the MNIST notebook.

You can see the MNISTViTModel model in the models.py module, with an example of usage in the MNIST ViT notebook.

You can also try the generated model yourself by running the following command in the src/projects/mnist directory:

python canvas.py

python -m pytest tests.py