hookeai.model_architectures.rc_base_model.train.training.save_best_parameters

save_best_parameters(model, best_model_parameters, model_parameters_bounds)

Save best performance state model parameters.

Best performance state model parameters file is stored in model_directory under the name parameters_best.pkl.

Overwrites existing model parameters file.

Parameters:

• model (torch.nn.Module) – Model.

• best_model_parameters (dict) – Model best performance state learnable parameters. For each model parameter (key, str), store the corresponding value (item, float).

• model_parameters_bounds (dict) – Model learnable parameters bounds. For each parameter (key, str), the corresponding bounds are stored as a tuple(lower_bound, upper_bound) (item, tuple).