mdlearn.nn.models.vde.model

Warning

VDE models are still under development, use with caution!

Classes

VDE(*args, **kwargs)

Variational dynamics encoder base class module based off the "Variational Encoding of Complex Dynamics" paper Inherits from mdlearn.nn.models.vae.VAE.

class mdlearn.nn.models.vde.model.VDE(*args: Any, **kwargs: Any)

Variational dynamics encoder base class module based off the “Variational Encoding of Complex Dynamics” paper Inherits from mdlearn.nn.models.vae.VAE.

ac_loss(z_t: torch.Tensor, z_t_tau: torch.Tensor) → torch.Tensor

Negative autocorrelation loss.

Parameters

• z_t (torch.Tensor) – \(z_t\)-latent vector.

• z_t_tau (torch.Tensor) – \(z_{t+\tau}\)-latent vector.

Returns

torch.Tensor – Negative autocorrelation loss between \(z_t\) and \(z_{t+\tau}\).