mlrl.testbed_sklearn.experiments.prediction.extension module¶

Provides classes that allow configuring the functionality to obtain predictions from a machine learning model.

class mlrl.testbed_sklearn.experiments.prediction.extension.PredictionTypeExtension(*dependencies: Extension)¶

An extension that configures the type of predictions to be obtained from a machine learning model.

static get_prediction_type(args: Namespace) → PredictionType¶

Returns the PredictionType to be used for obtaining predictions from a machine learning model according to the configuration.

get_supported_modes() → set[ExperimentMode]¶: See mlrl.testbed.extensions.extension.Extension.get_supported_modes()