Кероване навчання зі scikit-learn
George Boorman
Core Curriculum Manager
Регресія Ridge/Lasso: вибір alpha
KNN: вибір n_neighbors
Гіперпараметри: параметри, які задаємо до навчання моделі
alpha і n_neighborsСпробуйте багато значень гіперпараметрів
Навчіть модель для кожного з них окремо
Оцініть якість
Оберіть найкращі значення
Це називається налаштуванням гіперпараметрів
Обов'язково використовуйте кросвальдацію, щоб не підганятись під тестову вибірку
Дані все одно ділимо й кросвальдуємо на тренувальній частині
Тестову вибірку тримаємо для фінальної оцінки



from sklearn.model_selection import GridSearchCVkf = KFold(n_splits=5, shuffle=True, random_state=42)param_grid = {"alpha": np.arange(0.0001, 1, 10), "solver": ["sag", "lsqr"]}ridge = Ridge()ridge_cv = GridSearchCV(ridge, param_grid, cv=kf)ridge_cv.fit(X_train, y_train)print(ridge_cv.best_params_, ridge_cv.best_score_)
{'alpha': 0.0001, 'solver': 'sag'}
0.7529912278705785
from sklearn.model_selection import RandomizedSearchCVkf = KFold(n_splits=5, shuffle=True, random_state=42) param_grid = {'alpha': np.arange(0.0001, 1, 10), "solver": ['sag', 'lsqr']} ridge = Ridge()ridge_cv = RandomizedSearchCV(ridge, param_grid, cv=kf, n_iter=2) ridge_cv.fit(X_train, y_train)print(ridge_cv.best_params_, ridge_cv.best_score_)
{'solver': 'sag', 'alpha': 0.0001}
0.7529912278705785
test_score = ridge_cv.score(X_test, y_test)print(test_score)
0.7564731534089224
Кероване навчання зі scikit-learn