Skip to content

learner

learner

Module for learner

Classes

Learner

Bases: ABC

Wrapper to do the training

Functions
run_training abstractmethod 
run_training(
    exp_context,
    model_factory,
    train_set,
    validation_set,
    train_params,
    data_description,
)

Runs the training

Source code in niceml/mlcomponents/learners/learner.py 
15
16
17
18
19
20
21
22
23
24
25
@abstractmethod
def run_training(  # noqa: PLR0913
    self,
    exp_context: ExperimentContext,
    model_factory: ModelFactory,
    train_set: Dataset,
    validation_set: Dataset,
    train_params: TrainParams,
    data_description: DataDescription,
):
    """Runs the training"""