Xgboost: Quantile Regression and Support for Prediction Intervals

Created on 2 Aug 2016  ·  3Comments  ·  Source: dmlc/xgboost

I know that sklearn.ensemble.GradientBoostingRegressor supports quantile regression and the production of prediction intervals. Are there any plans for the XGBoost package to offer similar support?

feature-request

Most helpful comment

Quantile regression with xgboost requires custom gradient and hessian functions. Here is an implementation in Python: http://www.bigdatarepublic.nl/regression-prediction-intervals-with-xgboost/

All 3 comments

Quantile regression is currently not supported.

It can be achieved by defining the objective function on user's side: https://github.com/dmlc/xgboost/blob/master/R-package/demo/custom_objective.R

Quantile regression with xgboost requires custom gradient and hessian functions. Here is an implementation in Python: http://www.bigdatarepublic.nl/regression-prediction-intervals-with-xgboost/

Note that implementation is not very useful for most users, since it specifies grid-searching three parameters (very costly) to get a quantile estimate. In sklearn, you define only the quantile value and you are given a very robust and reliable estimate of that quantile.

Was this page helpful?
0 / 5 - 0 ratings

Related issues

colinsongf picture colinsongf  ·  4Comments

wenbo5565 picture wenbo5565  ·  3Comments

trivialfis picture trivialfis  ·  3Comments

nicoJiang picture nicoJiang  ·  4Comments

lizsz picture lizsz  ·  3Comments