Skip to content
No description, website, or topics provided.
Branch: master
Clone or download
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
output
.gitignore
LICENSE
MANIFEST.in
README.md
requirements.txt
setup.py

README.md

dtreeplt

it draws Decision Tree not using Graphviz, but only matplotlib.

Output Image using proposed method (using only matplotlib)

graphviz

Output Image using conventional method (Using Graphviz)

graphviz

Installation

pip install dtreeplt
Requirements are numpy(>=1.15.1), matplotlib(>=3.0.2) and scikit-learn(>=0.20.2) only.
Python 3.6.X.

Usage

Quick Start

from dtreeplt import dtreeplt
dtree = dtreeplt()
dtree.view()

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.