-
Notifications
You must be signed in to change notification settings - Fork 5.4k
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
[train] Storage: Update XGBoost + LightGBM callback APIs #38558
Conversation
Signed-off-by: Kai Fricke <kai@anyscale.com>
Signed-off-by: Kai Fricke <kai@anyscale.com>
Signed-off-by: Kai Fricke <kai@anyscale.com>
Signed-off-by: Kai Fricke <kai@anyscale.com>
Signed-off-by: Kai Fricke <kai@anyscale.com>
Signed-off-by: Kai Fricke <kai@anyscale.com>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks! Just some minor comments.
_checkpoint_callback_cls = None | ||
_report_callbacks_cls = None |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Remove these?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
We check if these properties are set for backwards compatibility with xgboost-ray and lightgbm-ray (and also for sequencing the API update).
We could test this also with getattr
but it makes the code that invokes them more cluttered.
Signed-off-by: Kai Fricke <kai@anyscale.com>
Signed-off-by: Kai Fricke <kai@anyscale.com>
Signed-off-by: Kai Fricke <kai@anyscale.com>
…#38558) The APIs are currently still using the deprecated `tune.checkpoint_dir` context manager and `tune.report`. This PR updates XGBoost+LightGBM to use `train.report` instead. It also consolidates the available callbacks into one. Signed-off-by: Kai Fricke <kai@anyscale.com> Signed-off-by: e428265 <arvind.chandramouli@lmco.com>
…#38558) The APIs are currently still using the deprecated `tune.checkpoint_dir` context manager and `tune.report`. This PR updates XGBoost+LightGBM to use `train.report` instead. It also consolidates the available callbacks into one. Signed-off-by: Kai Fricke <kai@anyscale.com> Signed-off-by: Victor <vctr.y.m@example.com>
Why are these changes needed?
The APIs are currently still using the deprecated
tune.checkpoint_dir
context manager andtune.report
.This PR updates XGBoost+LightGBM to use
train.report
instead. It also consolidates the available callbacks into one.Related issue number
Checks
git commit -s
) in this PR.scripts/format.sh
to lint the changes in this PR.method in Tune, I've added it in
doc/source/tune/api/
under thecorresponding
.rst
file.