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