GeoffChurch / BinaryDecisionTree

ID3 decision tree in Python with chi-squared pruning

BinaryDecisionTree

ID3 decision tree in Python with chi-squared pruning

Run this program with:
python tree.py --maxSamples 1000 --maxAttrs 50 --maxPValue 0.05 --maxEntropy 0.2
or
python tree.py --help for more options.

See the .ipynb file for more information!