get_expt_dir

behavenet.fitting.utils.get_expt_dir(hparams, model_class=None, model_type=None, expt_name=None)[source]

Get output directories associated with a particular model class/type/testtube expt name.

Examples

  • autoencoder: session_dir/ae/conv/08_latents/expt_name

  • arhmm: session_dir/arhmm/08_latents/16_states/stationary/gaussian/expt_name

  • arhmm-labels: session_dir/arhmm-labels/16_states/stationary/gaussian/expt_name

  • neural->ae decoder: session_dir/neural-ae/08_latents/mlp/mctx/expt_name

  • neural->arhmm decoder: session_dir/neural-ae/08_latents/16_states/stationary/mlp/mctx/expt_name

  • bayesian decoder: session_dir/arhmm-decoding/08_latents/16_states/stationary/gaussian/mctx/expt_name

Parameters:

  • hparams (dict) – specify model hyperparameters

  • model_class (str, optional) – will search hparams if not present

  • model_type (str, optional) – will search hparams if not present

  • expt_name (str, optional) – will search hparams if not present

Returns:

contains data info (lab/expt/animal/session) as well as model info (e.g. n_ae_latents) and expt_name

Return type:

str