kosmos.ml.cl_manager¶
Classes¶
- class CLManager(config: kosmos.ml.config.sl_train.SLTrainConfig)¶
Centralized learning manager for supervised learning classification tasks.
Initialize the centralized learning manager.
- Parameters:
config – Supervised learning training configuration.
Methods
- train() collections.abc.Iterator[kosmos.ml.sl_result.SLTrainIterationResult]¶
Run training on the train data over the configured number of epochs.
- Returns:
An iterator yielding one train result per epoch.
- Return type:
Iterator[SLTrainIterationResult]
- test() kosmos.ml.sl_result.SLTestIterationResult¶
Evaluate the model on the test data.
- Returns:
Result of the test iteration.
- Return type: