from enum import Enum
[docs]class StudyUserAttrs(str, Enum):
"""Dict keys for User Parameters of Optuna Study objects."""
OPTCONFIG = "optconfig"
[docs]class TrialUserAttrs(str, Enum):
"""Dict keys for User Parameters of Optuna Trial objects."""
TRAIN_SCORES = "train_scores"
TEST_SCORES = "test_scores"
[docs]class TrialParams(str, Enum):
"""Dict keys for Parameters of Optuna Trial objects."""
DESCRIPTOR = "descriptor"
ALGORITHM_NAME = "algorithm_name"
ALGORITHM_HASH = "algorithm_hash"
[docs]class MlflowLogParams(str, Enum):
"""Dict keys for Parameters of MLflow Logs."""
TRIAL_NUMBER = "trial_number"