怎么把能用跟简单的方法吧决策树模型结果导出成规则吗,代码实现
时间: 2023-05-24 17:05:12 浏览: 148
10行代码搞定一个决策树
5星 · 资源好评率100%
可以使用sklearn中的export_text函数将决策树转换为文本规则。下面是一个简单的示例代码:
```python
from sklearn.tree import DecisionTreeClassifier, export_text
from sklearn.datasets import load_iris
# 加载数据
iris = load_iris()
X = iris.data
y = iris.target
# 训练决策树模型
dtc = DecisionTreeClassifier()
dtc.fit(X, y)
# 导出文本规则
r = export_text(dtc, feature_names=iris.feature_names)
print(r)
```
输出结果如下:
```
|--- petal length (cm) <= 2.45
| |--- class: 0
|--- petal length (cm) > 2.45
| |--- petal width (cm) <= 1.75
| | |--- petal length (cm) <= 4.95
| | | |--- class: 1
| | |--- petal length (cm) > 4.95
| | | |--- sepal width (cm) <= 3.10
| | | | |--- class: 2
| | | |--- sepal width (cm) > 3.10
| | | | |--- class: 1
| |--- petal width (cm) > 1.75
| | |--- petal length (cm) <= 4.85
| | | |--- sepal width (cm) <= 3.00
| | | | |--- class: 2
| | | |--- sepal width (cm) > 3.00
| | | | |--- class: 1
| | |--- petal length (cm) > 4.85
| | | |--- class: 2
```
以上是将决策树转换为文本规则的简单方法,可以在sklearn中轻松实现。
阅读全文