python绘画决策树
时间: 2023-10-15 10:04:45 浏览: 50
你可以使用Python的多个库来绘制决策树,其中最常用的是scikit-learn和graphviz。下面我将给出使用这两个库的示例代码。
首先,确保你已经安装了这两个库。你可以使用以下命令来安装它们:
```python
pip install scikit-learn
pip install graphviz
```
然后,将下面的代码粘贴到Python文件中:
```python
from sklearn import datasets
from sklearn.tree import DecisionTreeClassifier
from sklearn import tree
import graphviz
# 加载示例数据集
iris = datasets.load_iris()
# 创建决策树分类器
clf = DecisionTreeClassifier(random_state=0)
clf.fit(iris.data, iris.target)
# 使用graphviz库将决策树可视化
dot_data = tree.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("decision_tree") # 将决策树保存为PDF或其他格式的文件
graph.view() # 在默认的图像查看器中显示决策树
```
运行这段代码后,你将在当前目录下看到一个名为"decision_tree"的文件,它是决策树的可视化结果。你也可以使用`graph.view()`在默认的图像查看器中直接显示决策树。
希望这可以帮助到你!如有其他问题,请随时提问。
相关推荐
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)