ngboost icon indicating copy to clipboard operation
ngboost copied to clipboard

Negative Binomial for regression

Open esmit61 opened this issue 3 years ago • 1 comments

Hi. I would like to suggest if there could be a Negative Binomial distribution for the regression tasks with NGBRegressor, i.e.,

ngb = NGBRegressor(Dist=NegativeBinomial).fit(X_train, Y_train)

esmit61 avatar Dec 07 '21 09:12 esmit61

@esmit61 You might want to use XGBoostLSS, that supports Negative Binomial Distribution Type I

https://github.com/StatMixedML/XGBoostLSS

StatMixedML avatar Dec 18 '21 09:12 StatMixedML