export_graphviz

sklearn.tree.export_graphviz(decision_tree, out_file=None, *, max_depth=None, feature_names=None, class_names=None, label='all', filled=False, leaves_parallel=False, impurity=True, node_ids=False, proportion=False, rotate=False, rounded=False, special_characters=False, precision=3, fontname='helvetica')[source]

Экспорт дерева решений в формате DOT.

Эта функция генерирует представление дерева решений в формате GraphViz, которое затем записывается в out_file. После экспорта графические изображения можно сгенерировать, например, с помощью:

$ dot -Tps tree.dot -o tree.ps      (PostScript format)
$ dot -Tpng tree.dot -o tree.png    (PNG format)

Числа образцов, которые отображаются, взвешиваются любыми весами образцов, которые могут присутствовать.

Подробнее см. в Руководстве пользователя.

Параметры:

decision_treeобъект

Модель дерева решений, которая должна быть экспортирована в GraphViz.

out_fileобъект или строка, по умолчанию None

Дескриптор или имя выходного файла. Если None, результат возвращается как строка.

Изменено в версии 0.20: Значение по умолчанию для out_file изменено с «tree.dot» на None.

max_depthцелое число, по умолчанию None

Максимальная глубина представления. Если None, дерево генерируется полностью.

feature_namesмассив-подобный объект формы (n_features,), по умолчанию None

Массив, содержащий имена признаков. Если None, будут использоваться общие имена («x[0]», «x[1]», …).

class_namesмассив-подобный объект формы (n_classes,) или bool, по умолчанию None

Имена каждой из целевых классов в порядке возрастания числового значения. Актуально только для классификации и не поддерживается для многовыходных задач. Если True, отображается символическое представление имени класса.

label{‘все’, ‘корень’, ‘нет’}, по умолчанию ‘все’

Показывать ли информативные метки для неопределённости и т. д. Доступные варианты: «все» для отображения на каждом узле, «корень» для отображения только в корневом узле, или «нет» для того, чтобы не отображать ни на каком узле.

filledbool, по умолчанию False

Если установлено значение True, узлы закрашиваются, чтобы указать преобладающий класс для классификации, экстремальность значений для регрессии или чистоту узла для многовыходных задач.

leaves_parallelbool, по умолчанию False

Если установлено True, все листья рисуются внизу дерева.

impuritybool, по умолчанию True

Если установлено True, отображается неопределённость на каждом узле.

node_idsbool, по умолчанию False

Если установлено значение True, отображается номер ID каждого узла.

proportionbool, по умолчанию False

Если установлено True, изменить отображение «значений» и/или «образцов» на пропорции и проценты соответственно.

rotatebool, по умолчанию False

Если установлено True, ориентировать дерево слева направо вместо сверху вниз.

roundedbool, по умолчанию False

Если установлено True, узлы рисуются с закругленными углами.

special_charactersbool, по умолчанию False

Если установлено False, игнорировать специальные символы для совместимости с PostScript.

precisionцелое число, по умолчанию 3

Количество знаков после запятой для чисел с плавающей точкой в значениях неопределённости, пороговых значений и атрибутов значений каждого узла.

fontnameстрока, по умолчанию ‘helvetica’

Имя шрифта, используемого для отображения текста.

Возвращаемое значение:

dot_dataстрока

Строковое представление входного дерева в формате GraphViz dot. Возвращается только если out_file равно None.

Добавлен в версии 0.18.

Примеры

>>> from sklearn.datasets import load_iris
>>> from sklearn import tree

>>> clf = tree.DecisionTreeClassifier()
>>> iris = load_iris()

>>> clf = clf.fit(iris.data, iris.target)
>>> tree.export_graphviz(clf)
'digraph Tree {...