DecisionTreeClassifier

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

СигнатураDecisionTreeClassifier(max_depth=None, criterion='gini', random_state=None)

Делит данные по признакам, минимизируя загрязнённость узлов (criterion='gini' или 'entropy'). Легко интерпретируется, но без ограничения max_depth склонно к переобучению.

from sklearn.tree import DecisionTreeClassifier

clf = DecisionTreeClassifier(max_depth=4, random_state=42)
clf.fit(X_train, y_train)
print(clf.score(X_test, y_test))

← Все записи: scikit-learn