nvflare.app_opt.xgboost.tree_based.model_persistor module

class XGBModelPersistor(save_name='xgboost_model.json', load_as_dict=True)[source]

Bases: ModelPersistor

Init FLComponent.

The FLComponent is the base class of all FL Components. (executors, controllers, responders, filters, aggregators, and widgets are all FLComponents)

FLComponents have the capability to handle and fire events and contain various methods for logging.

handle_event(event: str, fl_ctx: FLContext)[source]

Handles events.

Parameters

  • event_type (str) – event type fired by workflow.

  • fl_ctx (FLContext) – FLContext information.

load_model(fl_ctx: FLContext) → Learnable[source]

Initialize and load the Model.

Parameters

fl_ctx – FLContext

Returns

ModelLearnable object

save_model(model_learnable: Learnable, fl_ctx: FLContext)[source]

Persists the Model object.

Parameters

  • model_learnable – ModelLearnable object

  • fl_ctx – FLContext