Как изменить размер изображения дерева, используя sklearn tree и export_graph_viz в Jupyter Notebook

Question

Я использую export_graph_viz для визуализации дерева решений, но изображение в моем блокноте Jupyter выходит из поля зрения.

Если бы это был рисунок в виде диаграммы, я бы использовал команду plt.figure (figsize = (12,7)) ограничить визуализацию.Но в этом случае я не знаю, как действовать.

Ниже приведен снимок моего ноутбука Jupyter и то, что я вижу:

Answer 1

Вы можете сохранить визуализированное дерево в файл и затем отобразить его с помощью pyplot.

Пример:

import matplotlib.pyplot as plt
import pydotplus
import matplotlib.image as mpimg
import io

from sklearn.externals.six import StringIO
from sklearn.tree import export_graphviz

dot_data = io.StringIO()

export_graphviz(clf, out_file=dot_data, rounded=True, filled=True)

filename = "tree.png"
pydotplus.graph_from_dot_data(dot_data.getvalue()).write_png(filename)

plt.figure(figsize=(12,12))
img = mpimg.imread(filename)
imgplot = plt.imshow(img)

plt.show()

Результат: