mlrl.testbed_sklearn.experiments.prediction.predictor_global module

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

Provides classes for obtaining predictions from global machine learning models.

class mlrl.testbed_sklearn.experiments.prediction.predictor_global.GlobalPredictionFunction(learner: Any)

Bases: PredictionFunction

A function that obtains and returns global predictions from a learner.

class mlrl.testbed_sklearn.experiments.prediction.predictor_global.GlobalPredictor(prediction_type: PredictionType)

Bases: Predictor

Obtains predictions from a previously trained global model.

obtain_predictions(learner: Any, dataset: Any, dataset_type: DatasetType, **kwargs) → Generator[PredictionState, None, None]

See mlrl.testbed_sklearn.experiments.prediction.predictor.Predictor.obtain_predictions()