DecisionTreeRegressor

Дерево решений для регрессии.

СигнатураDecisionTreeRegressor(max_depth=None, random_state=None)

Строит дерево, разбивая пространство признаков на области с постоянным предсказанием. Параметр max_depth ограничивает глубину и борется с переобучением.

from sklearn.tree import DecisionTreeRegressor

model = DecisionTreeRegressor(max_depth=5, random_state=42)
model.fit(X_train, y_train)
print(model.score(X_test, y_test))

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