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
tune_hyperparameters() → None[source]

Tune the hyperparameters