决策树可视化相关配置:
软件环境配置:
首先需要按照可视化插件 graphviz ,网址 :Download | Graphviz
然后随便选择一个版本安装即可(不太建议安装最新版本,建议安装比最新版本低2-3个版本)。
然后下一步就安装好了。
后面我们手动添加环境变量即可.
注意:添加到我们刚才按照目录下的bin目录为止。
然后我们检验是否配置成功:
打开cmd,输入
dot -v
python包安装:
pip install graphviz
conda install graphviz
from sklearn.tree import export_graphviz
import graphviz
# 生成Graphviz格式的决策树图形
dot_data = export_graphviz(clf, feature_names=vectorizer.feature_names_, class_names=clf.classes_, filled=True, rounded=True) #clf是训练出来的决策树模型
# 渲染决策树图形
graph = graphviz.Source(dot_data)
# # 保存决策树图形为PDF文件
graph.render('./show/decision_tree')