2.2. mbl.workflow.grid_search

Functions

mlflow_tracker(profile_name)
run(tracking_uri, experiment_name, ...[, ...])

Classes

GridSearch()
RandomHeisenbergFoldingTSDRGGridSearch()
RandomHeisenbergTSDRGGridSearch()

mbl.workflow.grid_search.run(tracking_uri, experiment_name, experiment, configs, resources_per_trial=None, token=None, tags=None, **kwargs)

Parameters

• tracking_uri (str) –

• experiment_name (str) –

• experiment (Callable) –

• configs (Dict) –

• resources_per_trial (Optional[Dict]) –

• token (Optional[str]) –

• tags (Optional[Dict[str, str]]) –

mbl.workflow.grid_search.mlflow_tracker(profile_name)

Parameters

profile_name (str) –

class mbl.workflow.grid_search.GridSearch

Bases: abc.ABC

static experiment(func)

Parameters

func (Callable) –

class mbl.workflow.grid_search.RandomHeisenbergTSDRGGridSearch

Bases: mbl.workflow.grid_search.GridSearch

class AthenaMetadata(profile_name: str = 'default', s3_path: str = 's3://many-body-localization/random_heisenberg/tsdrg', database: str = 'random_heisenberg', table: str = 'tsdrg')

Bases: object

Parameters

• profile_name (str) –

• s3_path (str) –

• database (str) –

• table (str) –

Return type

None

profile_name: str = 'default'

s3_path: str = 's3://many-body-localization/random_heisenberg/tsdrg'

database: str = 'random_heisenberg'

table: str = 'tsdrg'

__init__(profile_name='default', s3_path='s3://many-body-localization/random_heisenberg/tsdrg', database='random_heisenberg', table='tsdrg')

Parameters

• profile_name (str) –

• s3_path (str) –

• database (str) –

• table (str) –

Return type

None

static experiment(config)

Parameters

config (Dict[str, Union[int, float, str]]) –

classmethod to_s3_parquet(df)

Parameters

df (pandas.core.frame.DataFrame) –

class mbl.workflow.grid_search.RandomHeisenbergFoldingTSDRGGridSearch

Bases: mbl.workflow.grid_search.GridSearch

class AthenaMetadata(profile_name: str = 'default', s3_path: str = 's3://many-body-localization/random_heisenberg/folding_tsdrg', database: str = 'random_heisenberg', table: str = 'folding_tsdrg')

Bases: object

Parameters

• profile_name (str) –

• s3_path (str) –

• database (str) –

• table (str) –

Return type

None

profile_name: str = 'default'

s3_path: str = 's3://many-body-localization/random_heisenberg/folding_tsdrg'

database: str = 'random_heisenberg'

table: str = 'folding_tsdrg'

__init__(profile_name='default', s3_path='s3://many-body-localization/random_heisenberg/folding_tsdrg', database='random_heisenberg', table='folding_tsdrg')

Parameters

• profile_name (str) –

• s3_path (str) –

• database (str) –

• table (str) –

Return type

None

static experiment(config)

Parameters

config (Dict[str, Union[int, float, str]]) –

classmethod retrieve_energy_bounds(config)

Parameters

config (Dict[str, Union[int, float, str]]) –

Return type

Dict[str, Union[int, float, str]]

classmethod to_s3_parquet(df)

Parameters

df (pandas.core.frame.DataFrame) –