python决策树报错，奇怪的错误

[yesterday once more]

求求各位大神了！ from sklearn import datasets from sklearn import tree from sklearn.externals.six import StringIO import pydot import numpy as np iris = datasets.load_iris() # 加载Iris数据集 # 载入特征和标签集 x = [[1, 1], [1, 1], [1, 0], [0, 1], [0, 1]] x=np.array(x) y = ['1', '1', '0', '0', '0'] y=np.array(y) z = ['浮出水面', '有无脚蹼'] p = ['是鱼类', '否鱼类'] p=np.array(p) clf = tree.DecisionTreeClassifier() # 构建决策树，默认是gini指标 #clf = clf.fit(iris.data, iris.target) clf = tree.DecisionTreeClassifier() clf = clf.fit(x, y) dot_data = StringIO() tree.export_graphviz(clf, out_file=dot_data, feature_names=z, class_names=p, filled=True, rounded=True, special_characters=True) #tree.export_graphviz(clf, out_file=dot_data, feature_names=iris.feature_names, class_names=iris.target_names, filled=True, rounded=True, special_characters=True) #tree.export_graphviz(clf, out_file=r"tree.dot") #把这行代码放开可以生成决策树的文件 (graph,) = pydot.graph_from_dot_data(dot_data.getvalue()) graph.write_png('iris.png')

[源代码大师]

希望对你有帮助：https://blog.csdn.net/it_xiangqiang/category_10581430.html 希望对你有帮助：https://blog.csdn.net/it_xiangqiang/category_10768339.html