决策树三分类可视化决策树
时间: 2023-10-16 12:10:34 浏览: 104
分类决策树
可以使用Graphviz库来可视化决策树。首先需要将决策树转换为Graphviz格式,然后使用Graphviz库生成可视化图形。
以下是一个简单的示例代码:
```python
from sklearn.tree import DecisionTreeClassifier, export_graphviz
import graphviz
# 构建决策树模型
clf = DecisionTreeClassifier()
clf.fit(X, y)
# 将决策树转换为Graphviz格式
dot_data = export_graphviz(clf, out_file=None,
feature_names=X.columns,
class_names=['class1', 'class2', 'class3'],
filled=True, rounded=True,
special_characters=True)
# 生成可视化图形
graph = graphviz.Source(dot_data)
graph.render('decision_tree') # 保存为PDF文件
```
其中,X为特征矩阵,y为目标变量。
阅读全文