vae_xprize
==========

.. py:module:: vae_xprize


Classes
-------

.. autoapisummary::

   vae_xprize.SamplingLayer
   vae_xprize.VAE


Functions
---------

.. autoapisummary::

   vae_xprize._get_encoder
   vae_xprize._get_decoder


Package Contents
----------------

.. py:class:: SamplingLayer

   Bases: :py:obj:`tensorflow.keras.layers.Layer`


   .. py:method:: call(inputs)


.. py:function:: _get_encoder(input_shape, latent_dim)

.. py:function:: _get_decoder(output_shape, latent_dim)

.. py:class:: VAE(input_shape, latent_dim, beta_kl=1, beta_reconst=1, encoder=None, decoder=None, **kwargs)

   Bases: :py:obj:`tensorflow.keras.Model`


   .. py:attribute:: encoder


   .. py:attribute:: decoder


   .. py:attribute:: beta_kl


   .. py:attribute:: beta_reconst


   .. py:attribute:: total_loss_tracker


   .. py:attribute:: reconstruction_loss_tracker


   .. py:attribute:: kl_loss_tracker


   .. py:property:: metrics


   .. py:method:: call(inputs, training=None, mask=None)


   .. py:method:: train_step(data)


   .. py:method:: test_step(data)