Python package treeplot vizualizes a tree based on a randomforest or xgboost model.
Project description
treeplot
- treeplot is Python package
Contents
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
model = RandomForestClassifier(n_estimators=100, max_depth=2, random_state=0).fit(X, y)
# Make plot
ax = treeplot.plot(model)
# or alternatively 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
model = XGBClassifier(n_estimators=100, max_depth=2, random_state=0).fit(X, y)
# Make plot
ax = treeplot.plot(model)
# or alternatively 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.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
treeplot-0.1.1.tar.gz
(4.7 kB
view hashes)