mlrl.testbed_sklearn.experiments.input.dataset.preprocessors.extension module

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

Provides classes that allow configuring the functionality to preprocess tabular datasets.

class mlrl.testbed_sklearn.experiments.input.dataset.preprocessors.extension.PreprocessorExtension(*dependencies: Extension)

Bases: Extension

An extension that configures the functionality to preprocess tabular datasets.

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

static get_preprocessors(args: Namespace) → list[Preprocessor]

Returns the preprocessors to be used for preprocessing datasets according to the configuration.

Parameters:

args – The command line arguments specified by the user

Returns:

The preprocessors to be used

get_supported_modes() → set[ExperimentMode]

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