正文

python决策树图怎么看

小六 /27 浏览量
0225

要查看Python生成的决策树图像，你可以使用以下步骤：

1. 确保你已经使用Python的决策树库（如`sklearn`）生成了决策树模型。

2. 使用`export_graphviz`函数将决策树导出为DOT格式。

3. 使用Graphviz软件将DOT文件转换为图像文件。

 from sklearn.datasets import load_iris from sklearn.tree import DecisionTreeClassifier from sklearn.tree import export_graphviz import os 加载数据集 iris = load_iris（） X = iris.data y = iris.target 创建决策树模型 clf = DecisionTreeClassifier（） clf = clf.fit（X, y）  导出决策树为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） 将DOT数据写入文件 with open（"iris_decision_tree.dot", "w"） as f: f.write（dot_data） 使用Graphviz将DOT文件转换为PNG图像 os.system（"dot -Tpng iris_decision_tree.dot -o iris_decision_tree.png"）

执行上述代码后，你会在当前目录下得到一个名为`iris_decision_tree.png`的决策树图像文件。你可以使用任何图像查看器打开这个文件来查看决策树。

如果你没有安装Graphviz，你可以通过以下命令安装（以Ubuntu为例）：

 sudo apt-get install graphviz

安装完成后，你就可以按照上面的步骤生成并查看决策树图像了

-- 展开阅读全文 --