subaligner.hparam_tuner module¶
-
class
subaligner.hparam_tuner.
HyperParameterTuner
(av_file_paths: List[str], subtitle_file_paths: List[str], training_dump_dir: str, num_of_trials: int = 5, tuning_epochs: int = 5, network_type: str = 'lstm', **kwargs)[source]¶ Bases:
object
Hyperparameter tuning using the Tree of Parzen Estimators algorithm
-
SEARCH_SPACE
= {'back_hidden_size': hyperopt.hp.choice, 'batch_size': hyperopt.pyll.base.scope.int, 'dropout': hyperopt.hp.loguniform, 'front_hidden_size': hyperopt.hp.choice, 'learning_rate': hyperopt.hp.loguniform, 'optimizer': hyperopt.hp.choice, 'validation_split': hyperopt.hp.loguniform}¶
-
property
hyperparameters
¶
-