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_id and optionally additionally restricted by
get_trials(→ Dict[int, ...)	Gets TrialData for the given experiment_id and optionally

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_id and optionally additionally restricted by tunable_config_id.

The returned DataFrame includes each trial’s metadata, config, and results in wide format, with config parameters prefixed with ExperimentData.CONFIG_COLUMN_PREFIX and results prefixed with ExperimentData.RESULT_COLUMN_PREFIX.

See also

TunableConfigTrialGroupSqlData, 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_id and optionally additionally restricted by tunable_config_id.

See also

TunableConfigTrialGroupSqlData, 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]