Skip to content

Latest commit

 

History

History
16 lines (11 loc) · 2.01 KB

README.md

File metadata and controls

16 lines (11 loc) · 2.01 KB

mlem

A library where I put my implementations of whatever machine learning algorithms I've been fascinated by lately

Currently implemented

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:

GraphViz rendering of said tree

The name

mlem stands for Machine Learning Experiments Medley, and it's also a noise that cats sometimes make :3