modlee.retriever module

Retriever for experiment assets.

modlee.retriever.get_cached_vars(run_path)

Get the cached variables required to rebuild a model from a run path.

Parameters:

run_path – The run path.

Returns:

A dictionary of the cached variables.

modlee.retriever.get_data_snapshot(run_path)

Get the saved data snapshot from a run path.

Parameters:

run_path – The run path.

Returns:

The data snapshot as a numpy array.

modlee.retriever.get_model(run_path)

Get the model at a run path.

Parameters:

run_path – The run path.

Returns:

The model as a ModleeModel object.

modlee.retriever.get_runs(run_path, experiment_id=None, run_id=None, **kwargs)

Get the runs in a given run path.

Parameters:

• run_path – The path to search.

• experiment_id – The experiment ID to retrieve, defaults to None and retrieves all experiments.

• run_id – The run ID to retrieve, defaults to None to retrieves all runs.

Returns:

A list of runs.

modlee.retriever.run_path_exists(run_path)

Chek if a run path exists.

Parameters:

run_path – The run path to check.

Returns:

Whether the path exists or not.