export_graphviz

sklearn.tree.export_graphviz(decision_tree, out_file=None, *, max_depth=None, feature_names=None, class_names=None, метка='all', заполненный=False, leaves_parallel=False, impurity=True, node_ids=False, пропорция=False, вращать=False, округленный=False, special_characters=False, точность=3, fontname='helvetica')

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

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

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

Показанные количества образцов взвешены с учетом любых sample_weights, которые могут присутствовать.

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

Параметры:

decision_treeobject

Оценщик дерева решений для экспорта в GraphViz.

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

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

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

max_depthint, default=None

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

feature_namesarray-like формы (n_features,), по умолчанию=None

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

class_namesarray-like формы (n_classes,) или bool, по умолчанию=None

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

метка{‘all’, ‘root’, ‘none’}, по умолчанию=’all’

Показывать ли информативные метки для примеси и т.д. Варианты включают 'all' для показа в каждом узле, 'root' только в верхнем корневом узле или 'none' для отключения показа во всех узлах.

заполненныйbool, по умолчанию=False

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

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

При установке значения True, отображать все листовые узлы внизу дерева.

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

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

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

При установке значения True, показывать идентификационный номер на каждом узле.

пропорцияbool, по умолчанию=False

При установке значения True, измените отображение 'values' и/или 'samples' на пропорции и проценты соответственно.

вращатьbool, по умолчанию=False

При установке значения True, ориентировать дерево слева направо, а не сверху вниз.

округленныйbool, по умолчанию=False

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

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

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

точностьint, по умолчанию=3

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

fontnamestr, default='helvetica'

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

Возвращает:

dot_datastr

Строковое представление входного дерева в формате 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 {...