mlrl.testbed_sklearn.experiments.experiment module

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

Provides classes for performing experiments using the scikit-learn framework.

class mlrl.testbed_sklearn.experiments.experiment.SkLearnExperiment(initial_state: ExperimentState, dataset_splitter: DatasetSplitter)

Bases: Experiment

An experiment that trains and evaluates a machine learning model using the scikit-learn framework.

class Builder(initial_state: ExperimentState, dataset_splitter: DatasetSplitter)

Bases: Builder

Allows to configure and create instances of the class SkLearnExperiment.