pyraug.models.nn

In this module are stored the main Neural Networks Architectures.

• BaseEncoder
• BaseDecoder
• BaseMetric

The Base Architectures

BaseEncoder	This is a base class for Encoders neural networks.
BaseDecoder	This is a base class for Decoders neural networks.
BaseMetric	This is a base class for Metrics neural networks (only applicable for Riemannian based VAE)

Note

In case you want to provide your own neural architecture, make sure you make them inherit from these classes. See Making your own autoencoder model.