Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Native quantile regression for xgb 2.0.0 and above #2051

Merged
merged 4 commits into from
Nov 3, 2023

Conversation

dennisbader
Copy link
Collaborator

Fixes #1669

Summary

  • Leverage built-in Quantile Regression for XGBoost versions >= 2.0.0 to improve accuracy

Other Information

  • XGBoost supports multi quantile regression which we do not levarage yet. Instead, we train a separate model per quantile to use the same logic as with LightGBM. We can change this in the future

Comparison Quantile Regression with XGBoost < 2.0.0 (top) and XGBoost >= 2.0.0 (bottom)

image

image

@codecov-commenter
Copy link

Codecov Report

Attention: 1 lines in your changes are missing coverage. Please review.

❗ Your organization needs to install the Codecov GitHub app to enable full functionality.

Files Coverage Δ
darts/models/forecasting/xgboost.py 87.69% <85.71%> (-12.31%) ⬇️

... and 6 files with indirect coverage changes

📢 Thoughts on this report? Let us know!.

@dennisbader dennisbader merged commit 2d0233f into master Nov 3, 2023
9 checks passed
@dennisbader dennisbader deleted the feat/builtin_xgb_quantile_regression branch November 3, 2023 13:48
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[BUG] Constant Predictions for Multivariate Probabilistic XGBoost with Quantile Likelihood.
3 participants