A library where I put my implementations of whatever machine learning algorithms I've been fascinated by lately
Decision tree classifier
mlem.dectree.ID3Classifier
is an implementation of the ID3 algorithm with the extension of supporting any number of values for the target variable rather than just yes/no. There is a demo for it in tests/test_dectree.py
, which outputs the decision tree generated for the classic dataset tests/datasets/tennis.csv
.
The output looks like this on Graphviz Online:
mlem
stands for Machine Learning Experiments Medley, and it's also a noise that cats sometimes make :3