plot_tree#

sklearn.tree.plot_tree(decision_tree, *, max_depth=None, feature_names=None, class_names=None, метка='all', заполненный=False, impurity=True, node_ids=False, пропорция=False, округленный=False, точность=3, ax=None, размер шрифта=None)[источник]#

Постройте дерево решений.

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

Визуализация автоматически подгоняется под размер оси. Используйте figsize или dpi аргументы plt.figure для управления размером отрисовки.

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

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

Параметры:
decision_treeрегрессор или классификатор на основе дерева решений

Дерево решений для построения графика.

max_depthint, default=None

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

feature_namesarray-like из str, по умолчанию=None

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

class_namesмассивоподобный из str или True, по умолчанию=None

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

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

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

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

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

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

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

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

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

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

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

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

При установке значения True, рисуйте узлы с закруглёнными углами и используйте шрифты Helvetica вместо Times-Roman.

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

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

axось matplotlib, по умолчанию=None

Оси для построения графика. Если None, используется текущая ось. Любое предыдущее содержимое очищается.

размер шрифтаint, default=None

Размер шрифта текста. Если None, определяется автоматически, чтобы соответствовать рисунку.

Возвращает:
аннотациисписок художников

Список, содержащий художников для аннотационных блоков, составляющих дерево.

Примеры

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

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

>>> clf = clf.fit(iris.data, iris.target)
>>> tree.plot_tree(clf)
[...]
На этой странице