import numpy as np import matplotlib.pyplot as plt from sklearn import datasets #引入sklearn自带的数据集 from sklearn.tree import DecisionTreeClassifier #引入决策树分类模块 X,Y=[],[] #读取数据 fr = open("D:\大数据技术学习笔记\python学习\机器人学习\data\knn.txt") for line in fr.readlines(): line = line.strip().split() X.append([int(line[0]),int (line[1])]) Y.append(int(line[-1])) X=np.array(X) #转换成NumPy数组,x是特征属性集 #y是类别标签集 Y=np.array(Y) #去掉这3行的注释符即可对鸢尾花数据集分类 #iris = datasets.load_iris() #X = iris.data[:, [0, 2]] #Y = iris.target #训练决策树模型,限制树的最大深度为4 clf = DecisionTreeClassifier("entropy",max_depth=4) clf.fit(X, Y) #画分类界面图 x_min, x_max = X[:,0].min() - 1,X[:, 0].max() + 1 y_min, y_max=X[:, 1].min() -1, X[:,1].max()+ 1 xx, yy=np.meshgrid(np.arange(x_min, x_max, 0.1), np.arange(y_min, y_max,0.1)) Z = clf.predict(np.c_[xx.ravel(), yy.ravel ()]) Z = Z.reshape(xx.shape) plt.contourf(xx, yy, Z, alpha=0.3) plt.scatter(X[:, 0], X[:, 1], c=Y, alpha=1) plt.show() 如何生成决策树图代码实现
时间: 2023-12-04 18:05:21 浏览: 143
Matplotlib.pyplot 三维绘图的实现示例
生成决策树图需要使用Graphviz(一个开源的图形可视化软件)和pydotplus(一个Python模块,用于将Graphviz生成的DOT语言文件转换为图形)。以下是生成决策树图的代码实现:
```python
from sklearn.tree import export_graphviz
import pydotplus
dot_data = export_graphviz(clf, out_file=None, feature_names=['feature1', 'feature2'], class_names=['class1', 'class2'], filled=True, rounded=True, special_characters=True)
graph = pydotplus.graph_from_dot_data(dot_data)
graph.write_png('decision_tree.png')
```
其中,`feature_names`和`class_names`分别指定特征属性和类别标签的名称,`filled`和`rounded`用于美化输出的图形,`special_characters`用于处理特殊字符。执行上述代码后,会在当前目录下生成名为`decision_tree.png`的决策树图。
阅读全文