How to visualize decision tree it in python?

Question

I created a decision tree and tried to follow an answer(Visualizing decision tree in scikit-learn) to visualize it in python,but still don't work:

import pandas as pd
score_v2 = pd.read_csv("C:/TEST_RF_CSV_simple.csv",encoding = "cp950")

from sklearn import datasets
from sklearn.model_selection import train_test_split
from sklearn.neighbors import KNeighborsClassifier

score_X = score_v2
score_y = score_v2.buy_lf
X_train, X_test, y_train, y_test = train_test_split(
score_X, score_y, test_size=0.3)

from sklearn.tree import DecisionTreeClassifier
tree=DecisionTreeClassifier(criterion = 'entropy', max_depth=3, 
random_state=0)
tree.fit(X_train, y_train)
tree_1 = tree.fit(X_train, y_train)

from sklearn.tree import export_graphviz
dotfile = open("D:/dtree2.dot", 'w')
tree.export_graphviz(dtree, out_file = dotfile, feature_names = 
X.columns)
dotfile.close()

my error was:

AttributeError: 'DecisionTreeClassifier' object has no attribute 
'export_graphviz'

Could any masters help me with solving the problem?

Possible duplicate of [Python Decision Tree GraphViz](https://stackoverflow.com/questions/34144134/python-decision-tree-graphviz) — sync11, Nov 26 '18 at 09:48

Matthieu Brucher · Answer 1 · 2018-11-26T10:07:43.443

1

export_graphviz is a function of sklearn.tree, not from the classifier:

from sklearn.tree import export_graphviz
export_graphviz(tree, out_file=dotfile, feature_names=X.columns)

edited Nov 26 '18 at 10:07

answered Nov 26 '18 at 09:45

Matthieu Brucher

21,634
7
38
62

Thanks. If I want to work it, did you know how to refine it? – kate Nov 26 '18 at 10:07
Such as this question's picture.(https://stackoverflow.com/questions/31786347/how-to-make-python-decision-tree-more-understandable) – kate Nov 26 '18 at 10:12
Sorry, I don't understand your question. You get a dot file for graphviz out of this command. Then you can convert it to a png. What do you mean by "work it" and "refine it"? – Matthieu Brucher Nov 26 '18 at 10:14
"refine". The final goal is output the picture of decision tree. – kate Nov 26 '18 at 10:17
Use graphviz (dot, to be precise) to convert the dot to a png. Look at the duplicate. – Matthieu Brucher Nov 26 '18 at 10:19
Thanks. But our computer didn't install pydot2, did you have other way? – kate Nov 26 '18 at 10:21
Please, look at the duplicate, there are 3 different ways! – Matthieu Brucher Nov 26 '18 at 10:23
I tried this,still no... with open("classifier.txt", "w") as f: f = tree.export_graphviz(tree, out_file=f) – kate Nov 26 '18 at 10:32
You get a `dot` file, then you need to convert it. – Matthieu Brucher Nov 26 '18 at 10:34
Hi Matthieu, When I convert dot, still have error. I will input raw data and run again. And read other book.Thanks for helping. – kate Nov 26 '18 at 11:11

How to visualize decision tree it in python?

1 Answers1