AttributeError: модуль «pydot» не имеет атрибута «graph_from_dot_data» в Spyder

Question

Я пытаюсь выполнить следующий код:

from sklearn.datasets import load_iris 
from sklearn import tree 
import pydot 
clf = tree.DecisionTreeClassifier() 
iris = load_iris() 
clf = clf.fit(iris.data, iris.target) 
from sklearn.externals.six import StringIO 
from pydot import * 
dotfile = StringIO() 
tree.export_graphviz(clf, out_file = dotfile) 
pydot.graph_from_dot_data(dot_data.getvalue()).write_png("dtree2.png") 

и я получаю следующее сообщение об ошибке: AttributeError: модуль «pydot» не имеет атрибута 'graph_from_dot_data'

Я попытался найти решение, но не смог этого сделать. Пожалуйста, помогите мне в этом отношении.

Answer 1

1) Используйте pydotplus, если вы используете питона 3+

2) Измените последнюю строку pydotplus.graph_from_dot_data (dotfile.getvalue()). Write_png ("dtree2.png") в качестве переменной имя 'dotfile' и не 'dot_data'

PS - переустановить Graphviz после установки pydotplus

Надеется, что это помогает!

Answer 2

pydot.graph_from_dot_data() возвращает список, так что попробуйте:

graphs = pydot.graph_from_dot_data(dot_data.getvalue()) 
graphs[0].write_png("dtree2.png")