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')[исходный код]

Экспортирует дерево решений в формате 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_fileobject or str, default=None

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

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

max_depthint, default=None

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

feature_namesarray-like of shape (n_features,), default=None

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

class_namesarray-like of shape (n_classes,) or bool, default=None

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

label{‘all’, ‘root’, ‘none’}, default=’all’

Определяет, отображать ли информативные метки для impurity и т. д. Возможные значения: ‘all’ — отображать для каждого узла, ‘root’ — отображать только для корневого узла, ‘none’ — не отображать ни для одного узла.

filledbool, default=False

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

leaves_parallelbool, default=False

Если задано значение True, все листовые узлы располагаются в нижней части дерева.

impuritybool, default=True

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

node_idsbool, default=False

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

proportionbool, default=False

Если задано значение True, значения ‘values’ и/или ‘samples’ отображаются соответственно как доли и проценты.

rotatebool, default=False

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

roundedbool, default=False

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

special_charactersbool, default=False

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

precisionint, default=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 {...