kosmos.ml.fl.fl_manager¶
Classes¶
- class FLManager(config: kosmos.ml.config.sl_train.FLTrainConfig, client_nodes: list[kosmos.topology.node.Node], server_node: kosmos.topology.node.Node)¶
Federated learning manager for supervised learning classification tasks.
Initialize the federated learning manager.
- Parameters:
config (FLTrainConfig) – Federated learning training configuration.
client_nodes (list[Node]) – The nodes representing federated clients.
server_node (Node) – The node representing the federated server.
Methods
- train() collections.abc.Iterator[kosmos.ml.sl_result.SLTrainIterationResult]¶
Run federated training across all configured rounds.
- Returns:
- An iterator yielding one training result per epoch
for all rounds.
- Return type:
Iterator[SLTrainIterationResult]
- test() kosmos.ml.sl_result.SLTestIterationResult¶
Evaluate the global model on the test dataset.
- Returns:
The result of the global model evaluation.
- Return type: