it draws Decision Tree not using Graphviz, but only matplotlib.
If interactive == True
, it draws Interactive Decision Tree on Notebook.
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.
from dtreeplt import dtreeplt
dtree = dtreeplt()
dtree.view()
# If you want to use interactive mode, set the parameter like below.
# dtree.view(interactive=True)
# 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')