GridSearchCV
Полный перебор гиперпараметров с кросс-валидацией.
Сигнатура
GridSearchCV(estimator, param_grid, cv=5, scoring=None)Перебирает все комбинации параметров из param_grid, оценивая каждую кросс-валидацией. Лучшие параметры доступны в best_params_, а готовая модель — в best_estimator_.
from sklearn.model_selection import GridSearchCV
from sklearn.svm import SVC
grid = {'C': [0.1, 1, 10], 'kernel': ['rbf', 'linear']}
gs = GridSearchCV(SVC(), grid, cv=5)
gs.fit(X_train, y_train)
print(gs.best_params_)