The model architecture of UNAGI 
--------------------------------

The module contains the VAE model and the discriminator model.


.. automodule:: UNAGI.model
   :members:
   :undoc-members:
   :show-inheritance:

.. autosummary::
    :toctree: 

    UNAGI.model.models


Trainer of UNAGI
----------------

The modules to start the training of UNAGI.


.. automodule:: UNAGI.train
   :members:
   :undoc-members:
   :show-inheritance:

.. autosummary::
    :toctree: 
    :nosignatures:
    
    UNAGI.train.trainer.UNAGI_trainer
    UNAGI.train.runner.UNAGI_runner
    UNAGI.train.customized_elbo