我正在使用决策树分类器,并想使用 matplotlib 绘制树
我正在使用这个,但节点很小而且不清晰:
from sklearn import tree
import matplotlib.pyplot as plt
plt.figure(figsize=(15,15))
tree.plot_tree(model_dt_smote,filled=True)
最佳答案
您可以将 axe
传递给具有较大 figsize
的 tree.plot_tree
并设置较大的 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)
输出:
关于python - 增加决策树中节点的大小,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/69604461/