mlos_bench.storage.util module

Utility functions for the storage subsystem.

mlos_bench.storage.util.kv_df_to_dict(dataframe: DataFrame) → Dict[str, int | float | str | None]

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

Parameters:

dataframepandas.DataFrame

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