AI/Machine Learning

[ML] Xgboost Error 해결 방법

운호(Noah) 2021. 1. 4. 12:37

에러

WARNING: src/objective/regression_obj.cu:152: reg:linear is now deprecated in favor of reg:squarederror.

해결 방법

  • 회귀 모델 매개 변수 objective = 'reg:lineaqr' 을 'reg:squarederror'로 변경

  •   xg_reg = xgb.XGBRegressor(objective='reg:squarederror', colsample_bytree = 0.3, learning_rate = 0.1, max_depth = 5, alpha = 10, n_estimators = 10)