Рисование дерева решений с питоном

Question

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

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

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

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

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

Answer 1

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

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

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

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