如何用代码实现决策树分类以及决策树分类的可视化。
时间: 2023-07-24 18:41:05 浏览: 76
决策树分类预测(Matlab完整源码)
5星 · 资源好评率100%
好的,我可以给您提供一些Python实现决策树分类和可视化的代码示例。
1. 决策树分类的实现
下面是使用sklearn库中的DecisionTreeClassifier类实现决策树分类的示例代码:
```python
from sklearn.tree import DecisionTreeClassifier
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
# 载入数据集
iris = load_iris()
X, y = iris.data, iris.target
# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)
# 创建决策树分类器
clf = DecisionTreeClassifier(random_state=42)
# 训练模型
clf.fit(X_train, y_train)
# 预测结果
y_pred = clf.predict(X_test)
# 计算准确率
accuracy = clf.score(X_test, y_test)
print('Accuracy:', accuracy)
```
2. 决策树分类的可视化
下面是使用Graphviz库实现决策树分类可视化的示例代码:
```python
from sklearn.tree import export_graphviz
import graphviz
# 将决策树导出为DOT格式
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)
# 使用Graphviz库生成决策树分类图
graph = graphviz.Source(dot_data)
graph.render('iris_decision_tree')
```
上述代码中,将决策树导出为DOT格式后,通过Graphviz库生成决策树分类图,并将其保存为PNG或PDF等格式的文件。您可以根据需要修改图像的格式和保存路径。
希望这些示例代码对您有所帮助!
阅读全文