Skip to content

modelfactory

modelfactory

Module for ModelFactory

Classes

ModelFactory

Bases: ABC

ABC for model factories. Used to create the model before training

Functions
create_model abstractmethod
create_model(data_description)

Creates a model for training according to the data_description

Source code in niceml/mlcomponents/models/modelfactory.py
@abstractmethod
def create_model(self, data_description: DataDescription) -> Any:
    """Creates a model for training according to the data_description"""