victorsuciu / ml-library

A small machine learning library for educational purposes

Another Machine Learning Library

This is a small machine learning library. I am builing it to improve my understanding of the math and algorithms used in ML. In adition, it is meant as an educational tool for students and teachers. This library will be continually updated as I learn new things.

Who is this for?

This library is meant to help educators and students learn more about machine learning and neural networks. Users can make gif animations of the training process, as shown below, which helps make sense of what "learning" really looks like.

Currently, it is not ready for release, but feel free to download it and play around. You'd just have to install it as a package and run it in a virtual Python environment, otherwise imports would be broken. This stackoverflow post goes more into detail.

Example Animations

Multlayer perceptron with Adam optimizer over the moons dataset.

Multlayer perceptron with Adam optimizer over concentric circles dataset.