External library integrations for Ray Tune#

PyTorch Lightning (tune.integration.pytorch_lightning)#

TuneReportCallback([metrics, on])

PyTorch Lightning to Ray Tune reporting callback

TuneReportCheckpointCallback([metrics, ...])

PyTorch Lightning report and checkpoint callback

XGBoost (tune.integration.xgboost)#

TuneReportCallback([metrics, ...])

XGBoost to Ray Tune reporting callback

TuneReportCheckpointCallback([metrics, ...])

XGBoost report and checkpoint callback

LightGBM (tune.integration.lightgbm)#

TuneReportCallback([metrics, ...])

Create a callback that reports metrics to Ray Tune.

TuneReportCheckpointCallback([metrics, ...])

Creates a callback that reports metrics and checkpoints model.