python sklearn 画出决策树并保存为PDF

Dragon水魅

发布于 2026-01-23 14:53:13

920

下载Graphviz

进入官网下载并安装： https://graphviz.gitlab.io/_pages/Download/Download_windows.html 并将下列路径配置为环境变量：

D:\software\Graphviz\bin

在cmd中测试：

dot -version

python代码

import numpy as np
import pandas as pd
from sklearn import tree
import graphviz

# x，y是sklearn中需要拟合的数据
x = np.array(exam_train)
y = np.array(classes_train)

clf = tree.DecisionTreeClassifier(criterion='entropy', class_weight='balanced', max_depth=25)
clf = clf.fit(x, y)

dot_data = tree.export_graphviz(clf, out_file=None, feature_names=None, filled=True, rounded=True)  # 重要参数可定制
graph = graphviz.Source(dot_data)
graph.render(view=True, format="pdf", filename="decisiontree_pdf")

可以生成一张贼帅的决策树PDF：