Skip to content
No description, website, or topics provided.
Jupyter Notebook Python
Branch: master
Clone or download
Pull request Compare This branch is 3 commits behind nekoumei:master.
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
dtreeplt
example
output
.gitignore
LICENSE
MANIFEST.in
README.md
requirements.txt
setup.py

README.md

dtreeplt

it draws Decision Tree not using Graphviz, but only matplotlib.
If interactive == True, it draws Interactive Decision Tree on Notebook.

Output Image using proposed method: dtreeplt (using only matplotlib)

graphviz

Output Image using conventional method: export_graphviz (Using Graphviz)

graphviz

Output Image using dtreeplt Interactive Decision Tree

graphviz

Installation

If you want to use the latest version, please use them on git.

pip install git+https://github.com/nekoumei/dtreeplt.git

when it comes to update, command like below.

pip install git+https://github.com/nekoumei/dtreeplt.git -U

Requirements: see requirements.txt
Python 3.6.X.

Usage

Quick Start

from dtreeplt import dtreeplt
dtree = dtreeplt()
dtree.view()
# If you want to use interactive mode, set the parameter like below.
# dtree.view(interactive=True)

Using trained DecisionTreeClassifier

# You should prepare trained model,feature_names, target_names.
# in this example, use iris datasets.
from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier
from dtreeplt import dtreeplt

iris = load_iris()
model = DecisionTreeClassifier()
model.fit(iris.data, iris.target)

dtree = dtreeplt(
    model=model,
    feature_names=iris.feature_names,
    target_names=iris.target_names
)
fig = dtree.view()
#if you want save figure, use savefig method in returned figure object.
#fig.savefig('output.png')
You can’t perform that action at this time.