In [1]학습·평가하기
from sklearn.ensemble import GradientBoostingRegressor
gb_model = GradientBoostingRegressor(n_estimators=200, learning_rate=0.04, max_depth=3, random_state=42)
gb_model.fit(X_train, y_train)
gb_pred = gb_model.predict(X_test)
print(f"mae: {mean_absolute_error(y_test, gb_pred)}")
print(f"rmse: {root_mean_squared_error(y_test, gb_pred)}")
print(f"r2score: {r2_score(y_test, gb_pred)}")
Out[1] 출력 보기
mae: 9801.771527224508 rmse: 32554.568013668282 r2score: 0.8068895356385997