nvflare.app_opt.sklearn.sklearn_executor module

class SKLearnExecutor(learner_id: str, train_task='train')[source]

Bases: Executor

An Executor interface for scikit-learn Learner.

Parameters:

  • learner_id (str) – id pointing to the learner object

  • train_task (str, optional) – label to dispatch train task. Defaults to AppConstants.TASK_TRAIN.

execute(task_name: str, shareable: Shareable, fl_ctx: FLContext, abort_signal: Signal) → Shareable[source]

Executes a task.

Parameters:

  • task_name (str) – task name.

  • shareable (Shareable) – input shareable.

  • fl_ctx (FLContext) – fl context.

  • abort_signal (Signal) – signal to check during execution to determine whether this task is aborted.

Returns:

An output shareable.

finalize(fl_ctx: FLContext)[source]
handle_event(event_type: str, fl_ctx: FLContext)[source]

Handles events.

Parameters:

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

  • fl_ctx (FLContext) – FLContext information.

initialize(fl_ctx: FLContext)[source]
load_log_tracker()[source]
log_value(key, value, step)[source]
save_model_global(model: any) → None[source]
save_model_local(model: any) → None[source]
train(current_round, global_param, fl_ctx: FLContext) → Shareable[source]
validate(current_round, global_param, fl_ctx: FLContext) → Shareable[source]