mlrl.testbed.experiments.output.model.extension module

Author: Michael Rapp (michael.rapp.ml@gmail.com)

Provides classes that allow configuring the functionality to write models to one or several sinks.

class mlrl.testbed.experiments.output.model.extension.ModelOutputDirectoryExtension(*dependencies: Extension)

Bases: Extension

An extension that configures the directory to which models should be written.

configure_experiment(args: Namespace, experiment_builder: Builder, _: ExperimentMode)

See mlrl.testbed.extensions.extension.Extension.configure_experiment()

get_supported_modes() → set[ExperimentMode]

See mlrl.testbed.extensions.extension.Extension.get_supported_modes()

class mlrl.testbed.experiments.output.model.extension.ModelOutputExtension(*dependencies: Extension)

Bases: Extension

An extension that configures the functionality to write models to one or several sinks.

SAVE_MODELS = <mlrl.util.cli.BoolArgument object>

get_supported_modes() → set[ExperimentMode]

See mlrl.testbed.extensions.extension.Extension.get_supported_modes()