ray.train.xgboost.RayTrainReportCallback.get_model

classmethod RayTrainReportCallback.get_model(checkpoint: Checkpoint, filename: str = 'model.ubj') → xgboost.core.Booster

Retrieve the model stored in a checkpoint reported by this callback.

Parameters:

• checkpoint – The checkpoint object returned by a training run. The checkpoint should be saved by an instance of this callback.

• filename – The filename to load the model from, which should match the filename used when creating the callback.

Returns:

The model loaded from the checkpoint.