Используя sklearn, как мне найти глубину дерева решений?

Question

Я тренирую дерево решений с помощью sklearn.Когда я использую:

dt_clf = tree.DecisionTreeClassifier()

, параметр max_depth по умолчанию равен None.Согласно документации, если max_depth равно None, то узлы расширяются до тех пор, пока все листья не станут чистыми или пока все листья не будут содержать менее min_samples_split выборок.

После подбора моей модели, как мне найтичто такое max_depth на самом деле?Функция get_params() не помогает.После примерки get_params() он все равно говорит None.

Как я могу получить фактический номер для max_depth?

Документы: https://scikit -learn.org / stable / modules / generate / sklearn.tree.DecisionTreeClassifier.html

Answer 1

Доступ к max_depth для базового Tree объекта:

from sklearn import tree
X = [[0, 0], [1, 1]]
Y = [0, 1]
clf = tree.DecisionTreeClassifier()
clf = clf.fit(X, Y)
print(clf.tree_.max_depth)
>>> 1

Вы можете получить более доступные атрибуты из базового объекта дерева, используя:

help(clf.tree_)

К ним относятся max_depth, node_count и другие параметры нижнего уровня.