In [1]학습·평가하기
from sklearn.ensemble import RandomForestRegressor
rf_model = RandomForestRegressor(n_estimators=300, random_state=42)
rf_model.fit(X_train, y_train)
rf_pred = rf_model.predict(X_test)
print(f"mae: {mean_absolute_error(y_test, rf_pred)}")
print(f"rmse: {root_mean_squared_error(y_test, rf_pred)}")
print(f"r2score: {r2_score(y_test, rf_pred)}")
Out[1] 출력 보기
mae: 10490.408566666667 rmse: 34131.04093053302 r2score: 0.787733723465766