treeplot 0.1.5

Python package treeplot vizualizes a tree based on a randomforest or xgboost model.

treeplot

• treeplot is Python package to easily plot the tree derived from models such as decisiontrees, randomforest and xgboost.

Contents

• Installation
• Quick Start
• Contribute
• Maintainers
• License

Installation

• Install treeplot from PyPI (recommended). treeplot is compatible with Python 3.6+ and runs on Linux, MacOS X and Windows.
• It is distributed under the MIT license.

Quick Start

pip install treeplot

• Alternatively, install treeplot from the GitHub source:

git clone https://github.com/erdogant/treeplot.git
cd treeplot
python setup.py install

Import treeplot package

import treeplot

Example RandomForest:

# Load example dataset
X,y = treeplot.import_example()
# Learn model
from sklearn.ensemble import RandomForestClassifier
model = RandomForestClassifier(n_estimators=100, max_depth=2, random_state=0).fit(X, y)

# Make plot
ax = treeplot.plot(model)

# If you have more parameters to specify:
ax = treeplot.randomforest(model, export='pdf')

Example XGboost:

# Load example dataset
X,y = treeplot.import_example()
# Learn model
from xgboost import XGBClassifier
model = XGBClassifier(n_estimators=100, max_depth=2, random_state=0).fit(X, y)

# Make plot
ax = treeplot.plot(model)

# If you have more parameters to specify:
ax = treeplot.xgboost(model, plottype='vertical')

Maintainers

• Erdogan Taskesen, github: erdogant

Contribute

• Contributions are welcome.

Licence

See LICENSE for details.

Donation

• This work is created and maintained in my free time. Contributions of any kind are very appreciated. Sponsering is also possible.