利用决策树算法对Iris数据集构建决策树并可视化
时间: 2023-07-20 21:03:25 浏览: 169
机器学习与深度学习-通过决策树算法分类鸢尾花数据集iris求出错误率画出决策树并进行可视化(完整源码+文档)0.zip
好的,我可以为您提供一个示例代码。首先,需要安装scikit-learn库和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.render("iris") # 保存决策树图像
```
运行后会生成一个iris.pdf文件,即为决策树的可视化结果。
阅读全文