bluecast.config.base_classes

Module Contents

Classes

BaseClassExperimentTracker

Base class for the experiment tracker.

class bluecast.config.base_classes.BaseClassExperimentTracker

Bases: abc.ABC

Base class for the experiment tracker.

Enforces the implementation of the add_results and retrieve_results_as_df methods.

abstract add_results(experiment_id: int, score_category: Literal[simple_train_test_score, cv_score, oof_score], training_config: bluecast.config.training_config.TrainingConfig, model_parameters: Dict[Any, Any], eval_scores: float | int | None, metric_used: str, metric_higher_is_better: bool) → None

Add results to the ExperimentTracker class.

abstract retrieve_results_as_df() → pandas.DataFrame

Retrieve results from the ExperimentTracker class