创建PDF使用pydot

Question

我从Visualizing a Decision Tree - Machine Learning 进口numpy的下面的代码为NP 从sklearn.datasets从sklearn进口树导入load_iris

iris = load_iris() 
test_idx = [0, 50 , 100] 

train_target = np.delete(iris.target, test_idx) 
train_data = np.delete(iris.data, test_idx , axis=0) 

test_target = iris.target[test_idx] 
test_data = iris.data[test_idx] 

clf = tree.DecisionTreeClassifier() 
clf.fit(train_data, train_target) 

print(test_target) 
print(clf.predict(test_data)) 

#viz_code 
from sklearn.externals.six import StringIO 
import pydot 
dot_data = StringIO() 
tree.export_graphviz(clf, 
     out_file=dot_data, 
     feature_names = iris.feature_names, 
     class_names = iris.target_names, 
     filled = True, rounded = True, 
     impurity = False) 
graph = pydot.graph_from_dot_data(dot_data.getvalue()) 
graph.write_pdf("iris.pdf") 

我想在我的Python 3.5运行它，但我得到说图表是一个列表的错误。

Traceback (most recent call last): 
    File "Iris.py", line 31, in <module> 
    graph.write_pdf("iris.pdf") 
AttributeError: 'list' object has no attribute 'write_pdf' 
Press any key to continue . . . 

为什么这里是一个列表？

Answer 1

Pydot不能在Python3中工作。 可以使用Pydotplus（graph.write_pdf("iris.pdf") AttributeError: 'list' object has no attribute 'write_pdf'“）为python3代替pydot。

虽然，在YouTube上显示的代码是Python2。所以，这将是更好，如果你使用Python2。