Skip to main content

vame.util.model_util

load_model

def load_model(cfg: dict, model_name: str, fixed: bool = True) -> RNN_VAE

Load the VAME model.

Arguments:

  • cfg dict - Configuration dictionary.
  • model_name str - Name of the model.
  • fixed bool - Fixed or variable length sequences.

Returns:

  • RNN_VAE - Loaded VAME model.