kooplearn.torch

nn.SpectralContrastiveLoss()

Spectral contrastive loss based originally introduced by HaoChen et al. [1], and adopted for evolution operators in Turri et al. [2], Jeong et al. [3]

nn.VampLoss([schatten_norm, center_covariances])

Variational Approach for learning Markov Processes (VAMP) score by Wu and Noé [1].

nn.AutoEncoderLoss([alpha_rec, alpha_lin, ...])

Single-step Dynamic Autoencoder (DAE) loss introduced by Lusch et al. [1].

nn.EnergyLoss([grad_weight])

Energy-based loss function.

FeatureMapEmbedder(encoder[, decoder, device])

sklearn-style transformer wrapping a PyTorch encoder (and optional decoder).