Рисование дерева решений с питоном - PullRequest
0 голосов
/ 30 октября 2019

Я работаю над деревом решений для рисования с помощью python,

tree.plot_tree(clf.fit(X_train, y_train))
plt.suptitle("Decision surface of a decision tree using paired features")
plt.show()

, но когда я запускаю этот код, дерево отображается следующим образом:

Дерево решений

Есть ли какой-нибудь возможный способ сделать дерево нормальным?

1 Ответ

1 голос
/ 31 октября 2019

export_graphviz может быть хорошей альтернативой для вас.

Вот ноутбук Kaggle , который объясняет, как наилучшим образом использовать эту функциональность, предоставляемую sklearn.

Для используемой вами функции plot_tree вы можете попробовать изменить размер шрифта с помощью аргумента fontsize. Или вы также можете попробовать изменить размер фигуры, если вы еще этого не сделали, выполнив что-то вроде этого -

fig = matplotlib.pyplot.gcf()
fig.set_size_inches(x, y) #dimensions you want
...