python绘制关于鸢尾花的决策树
时间: 2023-10-17 10:06:47 浏览: 71
要绘制决策树,我们可以使用Python中的Graphviz库。首先,我们需要安装Graphviz和pydotplus库。可以使用以下命令进行安装:
```
pip install graphviz
pip install pydotplus
```
然后,我们需要导入必要的库和数据集:
```python
from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier, export_graphviz
import graphviz
import pydotplus
iris = load_iris()
X = iris.data
y = iris.target
```
接下来,我们可以创建一个决策树分类器并拟合数据:
```python
clf = DecisionTreeClassifier(random_state=0)
clf.fit(X, y)
```
最后,我们可以使用export_graphviz函数创建一个Graphviz格式的决策树,并使用pydotplus将其转换为PDF格式:
```python
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 = pydotplus.graph_from_dot_data(dot_data)
graph.write_pdf("iris.pdf")
```
现在,我们可以在当前目录下找到名为“iris.pdf”的文件,它包含绘制的决策树。
阅读全文