不使用sklearn库将决策树的结果可视化
时间: 2023-09-10 08:08:03 浏览: 66
要将决策树的结果可视化,可以使用Graphviz库。下面是一个简单的例子:
首先,需要安装Graphviz库,可以使用pip进行安装:
```
pip install graphviz
```
然后,可以使用以下代码生成决策树的可视化结果:
```python
from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier, export_graphviz
import graphviz
# 加载数据集
iris = load_iris()
# 创建决策树模型
clf = DecisionTreeClassifier()
# 拟合数据
clf.fit(iris.data, iris.target)
# 生成决策树可视化结果
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.view()
```
这段代码会生成一个名为“Source.gv.pdf”的PDF文件,里面包含了决策树的可视化结果。
阅读全文