diart.optim#

Module Contents#

Classes#

Optimizer

class diart.optim.Optimizer(pipeline_class, speech_path, reference_path, study_or_path, batch_size=32, hparams=None, base_config=None, do_kickstart_hparams=True, metric=None, direction='minimize')#
Parameters:

  • pipeline_class (type) –

  • speech_path (Union[Text, pathlib.Path]) –

  • reference_path (Union[Text, pathlib.Path]) –

  • study_or_path (Union[diart.audio.FilePath, optuna.Study]) –

  • batch_size (int) –

  • hparams (Optional[Sequence[diart.blocks.base.HyperParameter]]) –

  • base_config (Optional[diart.blocks.PipelineConfig]) –

  • do_kickstart_hparams (bool) –

  • metric (Optional[pyannote.metrics.base.BaseMetric]) –

  • direction (typing_extensions.Literal[minimize, maximize]) –

property best_performance#
property best_hparams#
_callback(study, trial)#
Parameters:

  • study (optuna.Study) –

  • trial (optuna.trial.FrozenTrial) –

objective(trial)#
Parameters:

trial (optuna.trial.Trial) –

Return type:

float

__call__(num_iter, show_progress=True)#
Parameters:

  • num_iter (int) –

  • show_progress (bool) –