nvflare.app_opt.xgboost.data_loader module¶

class XGBDataLoader[source]¶

Bases: ABC

property client_id¶

property data_split_mode¶

initialize(client_id: str, rank: int, data_split_mode: DataSplitMode = DataSplitMode.ROW)[source]¶

abstract load_data() → Tuple[DMatrix, DMatrix][source]¶

Loads data for xgboost.

Returns:: A tuple of train_data, validation_data

property rank¶