nvflare.app_opt.xgboost.histogram_based_v2.csv_data_loader module

class CSVDataLoader(folder: str)

Bases: XGBDataLoader

Reads CSV dataset and return XGB data matrix.

Note: if split mode is vertical, we assume the label owner is rank 0.

Parameters:

folder – Folder to find the CSV files

load_data()

Loads data for xgboost.

Returns:

A tuple of train_data, validation_data