graph.write_pdf(“iris.pdf”) AttributeError: 'list' object has no attribute 'write_pdf'

Question

My code is follow the class of machine learning of google.The two code are same.I don\'t know why it show error.May be the type of variable is error.But google\'s code is sa

Answer 1

To add all graphs for the number of your n_estimators you can do:

for i in range(0, n):  #n is your n_estimators number
    dot_data = StringIO()
    tree.export_graphviz(clf.estimators_[i], out_file=dot_data, feature_names=iris.feature_names,
                        class_names=iris.target_names, filled=True, rounded=True,
                        impurity=False)
    graph = pydotplus.graph_from_dot_data(dot_data.getvalue())
    graph.write_pdf("iris%s.pdf"%i)

you could also switch the line

graph = pydotplus.graph_from_dot_data(dot_data.getvalue())

for this one

(graph,) = pydot.graph_from_dot_data(dot_data.getvalue())
graph.write_pdf("iris.pdf")

and it would still work.