-
-
Notifications
You must be signed in to change notification settings - Fork 8.7k
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
Initial support for multi-output regression. #7309
Conversation
Hmm .. datatable is updated? |
4cac240
to
c890e0f
Compare
That's exciting. We'd be interested in a XGBoost4J wrapper too, and I might be able to write such a thing once you're happy with the native API. We currently fake this up by training a separate Booster for each output dimension. |
Let me try to extend it to multi-class. Update: Will implement it in a different PR. We need to extend the objective function and have more tests around the behavior of prediction. This will push XGBoost's tree grid into a cubic and needs some refactoring. But the interface established in this PR should be the same ( |
@Craigacp Thanks for the support. I will keep you posted on the progress. I think it's best we wait until having a good solution for multi-class and other possible applications (like prob-forecasting). I will work on it in this release cycle. |
Should I create a multi-output regressor and distinguish it from the normal regressor? |
2a128a2
to
469857a
Compare
Need to have some utilities from #7331 first. |
Add multi-output regressor. Change tests/demo Specialize multi output reg. Fixes.
994d9bc
to
5519ea9
Compare
Cross linking #7083 . Need to resolve the same issue with matrix meta info. |
TO-DO for initial support of multi-target model training.
Tests:
|
Most of the basic infrastructures are in place now. Need more thought on classification problem due to different number of classes per target. |
The initial support is merged in #7514 . |
@Craigacp I think for multi-target classification models, xgboost needs a new interface and potentially lots of refactoring. I will focus on regression for now. Thank you for joining the discussion and feel free to test the new feature. ;-) |
@trivialfis Is this contain support for all multi output regression ? @Craigacp were you able to write the wrapper for XGBoost4J ? |
I haven't looked into it. I've had less time to work on XGBoost related things for the past couple of years. |
I will follow up with R and dask implementations if this PR is approved.
Related: #2087 .
Todos: