python决策树图怎么看
时间: 2023-07-28 18:07:52 浏览: 48
Python中可以使用scikit-learn库来构建决策树模型,并使用graphviz库来可视化决策树。
首先,需要安装graphviz库,可以使用以下命令进行安装:
```
pip install graphviz
```
然后,使用以下代码来构建决策树模型并可视化:
```python
from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier
from sklearn.tree import export_graphviz
import graphviz
# 加载数据集
iris = load_iris()
X = iris.data
y = iris.target
# 构建决策树模型
clf = DecisionTreeClassifier()
clf.fit(X, y)
# 可视化决策树
dot_data = export_graphviz(clf, 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.render("iris_decision_tree")
```
其中,`export_graphviz`函数可以将决策树模型导出为Graphviz格式,然后使用`graphviz.Source`函数将Graphviz格式的文件可视化。最后,使用`graph.render`函数将可视化结果保存为PDF或PNG格式的文件。