我正在使用决策树分类器，并希望使用matplotlib绘制树我正在使用这个，但节点很小，不清楚：
from sklearn import tree import matplotlib.pyplot as plt plt.figure(figsize=(15,15)) tree.plot_tree(model_dt_smote,filled=True)
roqulrg31#
您可以将axe传递给tree.plot_tree，其中figsize较大，并设置较大的fontsize，如下所示：
axe
tree.plot_tree
figsize
fontsize
from sklearn.datasets import load_iris import matplotlib.pyplot as plt from sklearn import tree clf = tree.DecisionTreeClassifier(random_state=0) iris = load_iris() clf = clf.fit(iris.data, iris.target) fig, axe = plt.subplots(figsize=(20,10)) tree.plot_tree(clf, ax = axe, fontsize=15)
输出：
1条答案
按热度按时间roqulrg31#
您可以将
axe传递给tree.plot_tree，其中figsize较大，并设置较大的fontsize，如下所示：输出：