Giters

stanfordmlgroup / ngboost

Natural Gradient Boosting for Probabilistic Prediction

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Use XGBoost as base estimator

fif911 opened this issue · comments

commented

First of all thanks for such a great library and your continuous support. I am working on predictions with uncertainty using the NGBoost Regressor and want to improve the accuracy of predictions. My assumption is in the base estimator is better the whole algorithm may perform better.

Do you think it's possible to use a more advanced algorithm than the DecisionTreeRegressor e.g. XGBoost or even NN?

Best regards

commented

See xgboost distribution! I have found that ngboost tended outperform it slightly, but maybe you can find better results. it's also up to 10x faster