GradientBoostingClassifier

Градиентный бустинг над деревьями.

СигнатураGradientBoostingClassifier(n_estimators=100, learning_rate=0.1, max_depth=3)

Последовательно строит деревья, каждое из которых исправляет ошибки предыдущих. Параметры learning_rate и n_estimators подбираются совместно. Часто даёт высокую точность.

from sklearn.ensemble import GradientBoostingClassifier

clf = GradientBoostingClassifier(learning_rate=0.05, n_estimators=200)
clf.fit(X_train, y_train)
print(clf.score(X_test, y_test))
← Все записи: scikit-learn
Поддержать проект