mlos_bench.storage.util

Utility functions for the storage subsystem.

Functions

kv_df_to_dict(→ Dict[str, ...)

Utility function to convert certain flat key-value dataframe formats used by the

Module Contents

mlos_bench.storage.util.kv_df_to_dict(dataframe: pandas.DataFrame) → Dict[str, mlos_bench.tunables.tunable.TunableValue | None]

Utility function to convert certain flat key-value dataframe formats used by the mlos_bench.storage modules to a dict.

Parameters:

dataframe (pandas.DataFrame) – A dataframe with exactly two columns, ‘parameter’ (or ‘metric’) and ‘value’, where ‘parameter’ is a string and ‘value’ is some TunableValue or None.

Return type:

Dict[str, Optional[mlos_bench.tunables.tunable.TunableValue]]