In [1]GridSearchCV로 alpha 고르고 평가하기
from sklearn.linear_model import Ridge
from sklearn.model_selection import GridSearchCV
r_model = Ridge()
param_grid = {'alpha': [0.001, 0.01, 0.1, 1, 10, 100]}
r_grid = GridSearchCV(r_model, param_grid, cv=5)
r_grid.fit(X_train, y_train)
print("best alpha:", r_grid.best_params_)
r_pred = r_grid.best_estimator_.predict(X_test)
print(f"mae: {mean_absolute_error(y_test, r_pred)}")
print(f"rmse: {root_mean_squared_error(y_test, r_pred)}")
print(f"r2score: {r2_score(y_test, r_pred)}")
Out[1] 출력 보기
best alpha: {'alpha': 100}
mae: 13022.236954337488
rmse: 28069.13123294435
r2score: 0.8564378755104337