dataeval.utils.models

Simple PyTorch model architectures used by DataEval.

Classes

AE

An autoencoder model with a separate encoder and decoder.

Decoder

A simple decoder to be used in an autoencoder model.

Encoder

A simple encoder to be used in an autoencoder model.

GMMDensityNet

Gaussian Mixture Model (GMM) density network for converting latent representations.

VAE

Variational Autoencoder model with separate encoder and decoder.

VAEDecoder

Variational Autoencoder Decoder.

VAEEncoder

Variational Autoencoder Encoder that outputs mean and log-variance.