决策树画图
时间: 2023-07-20 11:19:00 浏览: 144
基于梯度决策树+SVM的人体动作识别,基于梯度决策树的+lda的人体动作识别,基于梯度决策树的人体动作识别,基于SVM的人体动作
5星 · 资源好评率100%
要画出决策树,可以使用Python中的Graphviz库。以下是一个简单的例子:
首先,需要安装Graphviz库和pydotplus库:
```
pip install graphviz
pip install pydotplus
```
然后,可以使用以下代码创建一个简单的决策树:
```python
from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier, export_graphviz
from sklearn.externals.six import StringIO
import pydotplus
iris = load_iris()
X = iris.data[:, :2]
y = iris.target
clf = DecisionTreeClassifier(max_depth=3)
clf.fit(X, y)
dot_data = StringIO()
export_graphviz(clf, out_file=dot_data,
filled=True, rounded=True,
special_characters=True)
graph = pydotplus.graph_from_dot_data(dot_data.getvalue())
graph.write_png('iris.png')
```
这个例子使用了鸢尾花数据集,创建了一个最大深度为3的决策树,并将其保存为PNG文件。
在这个例子中,我们使用了export_graphviz函数从决策树中导出Graphviz格式的数据,并将其存储在dot_data变量中。然后,我们使用pydotplus库将这些数据转换为图形,并将其保存为PNG文件。
这只是一个简单的例子,对于更复杂的决策树,您可能需要使用更高级的技术来调整其外观和布局。
阅读全文