Package for removing the black-box around decision trees.
Inspired by Scikit-Learn's webpage on the matter, which you can find here:
http://scikit-learn.org/stable/auto_examples/tree/plot_unveil_tree_structure.html
The library aims to support all decision tree classes in Scikit-Learn. Currently, we support:
-
DecisionTreeClassifier
-
DecisionTreeRegressor
-
ExtraTreeClassifier
-
ExtraTreeRegressor
-
RandomForestClassifier
-
RandomForestRegressor
-
ExtraTreesClassifier
-
ExtraTreesRegressor
The code is available on PyPI and can be installed via pip:
pip install tree_decode
You can install the code from source by downloading the repository and running:
python setup.py install
After installation, you can run tests but starting up an interactive Python shell and running:
import tree_decode as td
td.test()Make sure to have pytest>=3.0 installed for testing purposes.
To see the code in action, you can find the demo by starting up an interactive Python shell and running:
import tree_decode as td
td.demo()