mlos_bench.storage.sql.common

Common SQL methods for accessing the stored benchmark data.

Functions

get_results_df(→ pandas.DataFrame)	Gets TrialData for the given experiment_data and optionally additionally restricted
get_trials(→ Dict[int, ...)	Gets TrialData for the given experiment_data and optionally additionally restricted

Module Contents

mlos_bench.storage.sql.common.get_results_df(engine: sqlalchemy.engine.Engine, schema: mlos_bench.storage.sql.schema.DbSchema, experiment_id: str, tunable_config_id: int | None = None) → pandas.DataFrame

Gets TrialData for the given experiment_data and optionally additionally restricted by tunable_config_id.

Used by both TunableConfigTrialGroupSqlData and ExperimentSqlData.

Parameters:

• engine (sqlalchemy.engine.Engine)

• schema (mlos_bench.storage.sql.schema.DbSchema)

• experiment_id (str)

• tunable_config_id (Optional[int])

Return type:

pandas.DataFrame

mlos_bench.storage.sql.common.get_trials(engine: sqlalchemy.engine.Engine, schema: mlos_bench.storage.sql.schema.DbSchema, experiment_id: str, tunable_config_id: int | None = None) → Dict[int, mlos_bench.storage.base_trial_data.TrialData]

Gets TrialData for the given experiment_data and optionally additionally restricted by tunable_config_id.

Used by both TunableConfigTrialGroupSqlData and ExperimentSqlData.

Parameters:

• engine (sqlalchemy.engine.Engine)

• schema (mlos_bench.storage.sql.schema.DbSchema)

• experiment_id (str)

• tunable_config_id (Optional[int])

Return type:

Dict[int, mlos_bench.storage.base_trial_data.TrialData]