DecisionTreeClassifier训练完成后,如何输出规则
时间: 2024-03-21 18:38:54 浏览: 43
在sklearn中,DecisionTreeClassifier的规则可以通过export_graphviz函数输出为Graphviz可视化文件格式。您可以按照以下步骤输出规则:
1. 导入需要的库:
```
from sklearn.tree import export_graphviz
import graphviz
```
2. 使用export_graphviz函数输出可视化文件:
```
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)
```
其中clf为已经训练好的DecisionTreeClassifier模型,iris为数据集,feature_names为特征名,class_names为分类名。
3. 使用graphviz将可视化文件转换为图像:
```
graph = graphviz.Source(dot_data)
graph.render("iris")
```
其中"iris"为保存的文件名,您可以根据需要更改。
4. 可以在当前工作目录下找到生成的iris.pdf文件,其中包含了决策树的规则。
阅读全文