给出一个使用graphviz可视化决策树模型的python代码
时间: 2024-05-04 08:15:45 浏览: 107
Python实现决策树并且使用Graphvize可视化的例子
5星 · 资源好评率100%
以下是一个使用graphviz可视化决策树模型的Python代码:
```python
from sklearn.tree import DecisionTreeClassifier, export_graphviz
from sklearn.datasets import load_iris
import graphviz
# 导入鸢尾花数据集
iris = load_iris()
# 构建决策树模型
dtree = DecisionTreeClassifier()
dtree.fit(iris.data, iris.target)
# 可视化决策树模型
dot_data = export_graphviz(dtree, out_file=None,
feature_names=iris.feature_names,
class_names=iris.target_names,
filled=True, rounded=True,
special_characters=True)
graph = graphviz.Source(dot_data)
graph
```
注意:此代码仅供参考,具体实现还需要根据具体情况进行调整。
阅读全文