mlrl.common.testbed.experiments.experiment module

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

Provides classes for performing experiments using rule learning algorithms.

class mlrl.common.testbed.experiments.experiment.RuleLearnerExperiment(args: Namespace, initial_state: ExperimentState, dataset_splitter: DatasetSplitter, training_procedure: TrainingProcedure | None = None, prediction_procedure: PredictionProcedure | None = None)

Bases: SkLearnExperiment

An experiment that trains and evaluates a rule 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 RuleLearnerExperiment.

class TrainingProcedure(base_learner: sklearn.base.BaseEstimator, fit_kwargs: dict[str, Any] | None = None)

Bases: TrainingProcedure

Allows to fit a rule learner to a training dataset.