调用sklearn.tree.plot_tree函数绘制决策树
时间: 2023-06-02 21:03:14 浏览: 207
Plot Ellipses:绘制椭圆-matlab开发
以下是一个示例代码,可以使用sklearn.tree.plot_tree函数绘制决策树:
```python
from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier
from sklearn.tree import plot_tree
# 加载鸢尾花数据集
iris = load_iris()
X = iris.data
y = iris.target
# 创建决策树分类器
clf = DecisionTreeClassifier(random_state=0)
clf.fit(X, y)
# 绘制决策树
plot_tree(clf)
```
该代码加载了鸢尾花数据集,创建了一个决策树分类器,然后使用plot_tree函数绘制了决策树。输出的图像将显示在jupyter或者ipython notebook中。
阅读全文