baby.training.v1_hyper_parameter_trainer.HyperParamV1¶
- class baby.training.v1_hyper_parameter_trainer.HyperParamV1(save_dir: Path, cnn_set, *args, **kwargs)¶
Bases:
object
Class that chooses the best hyperparameters for a specific model-type.
- Attributes
- best_parameters
- cnn
- cnn_dir
Methods
search
(**kwargs)Dummy search that just saves the default parameters to file.
save_best_parameters
- __init__(save_dir: Path, cnn_set, *args, **kwargs)¶
Methods
__init__
(save_dir, cnn_set, *args, **kwargs)save_best_parameters
(filename)search
(**kwargs)Dummy search that just saves the default parameters to file.
Attributes
best_parameters
cnn
cnn_dir