matchcake.ml.parameters_initialisation_strategies package¶

Submodules¶

matchcake.ml.parameters_initialisation_strategies.linear_strategy module¶

class matchcake.ml.parameters_initialisation_strategies.linear_strategy.LinearStrategy¶

Bases: ParametersInitialisationStrategy

NAME: str = 'Linear'¶

OPTIONAL_HYPERPARAMETERS = ['noise']¶

REQUIRES_HYPERPARAMETERS = ['n_layers', 'seed', 'parameters_rng']¶

__init__()¶

add_parameters_to_memory(parameters)¶

get_next_parameters(step_id: int, **hyperparameters)¶

initialise_parameters(**hyperparameters)¶

matchcake.ml.parameters_initialisation_strategies.parameters_initialisation_strategy module¶

class matchcake.ml.parameters_initialisation_strategies.parameters_initialisation_strategy.ParametersInitialisationStrategy¶

Bases: ABC

NAME: str = 'ParametersInitialisationStrategy'¶

OPTIONAL_HYPERPARAMETERS = []¶

REQUIRES_HYPERPARAMETERS = []¶

__init__()¶

add_parameters_to_memory(parameters)¶

check_required_hyperparameters(hyperparameters)¶

abstract get_next_parameters(step_id: int, **hyperparameters)¶

abstract initialise_parameters(**hyperparameters)¶

set_optional_hyperparameters(hyperparameters, default=None)¶

matchcake.ml.parameters_initialisation_strategies.random_strategy module¶

class matchcake.ml.parameters_initialisation_strategies.random_strategy.RandomStrategy¶

Bases: ParametersInitialisationStrategy

NAME: str = 'Random'¶

REQUIRES_HYPERPARAMETERS = ['current_named_parameters', 'parameters_rng', 'seed']¶

add_parameters_to_memory(parameters)¶

get_next_parameters(step_id: int, **hyperparameters)¶

initialise_parameters(**hyperparameters)¶

Module contents¶

matchcake.ml.parameters_initialisation_strategies.get_parameters_initialisation_strategy(name: str | None) → ParametersInitialisationStrategy¶