mlrl.testbed_sklearn.experiments.output.dataset.dataset_prediction module

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

Provides classes for representing predictions that are part of output data.

class mlrl.testbed_sklearn.experiments.output.dataset.dataset_prediction.PredictionDataset(dataset: TabularDataset)

Bases: TabularOutputDataset

Represents predictions for tabular data that are part of output data.

CONTEXT = Context(include_dataset_type=True, include_prediction_scope=True, include_fold=True)
PROPERTIES = Properties(name='Predictions', file_name='predictions')