AtomAI Models
==============

Segmentor
---------
.. autoclass:: atomai.models.Segmentor
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

ImSpec
-------
.. autoclass:: atomai.models.ImSpec
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

ParticleAnalyzer
----------------
.. autoclass:: atomai.models.ParticleAnalyzer
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

Denoiser
--------
.. autoclass:: atomai.models.DenoisingAutoencoder
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

Variational Autoencoder (VAE)
-----------------------------
.. autoclass:: atomai.models.VAE
    :members:
    :inherited-members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

Rotational Variational Autoencoder (rVAE)
------------------------------------------
.. autoclass:: atomai.models.rVAE
    :members:
    :inherited-members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

Joint Variational Autoencoder (jVAE)
------------------------------------------
.. autoclass:: atomai.models.jVAE
    :members:
    :inherited-members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

Joint Rotational Variational Autoencoder (jrVAE)
------------------------------------------------
.. autoclass:: atomai.models.jrVAE
    :members:
    :inherited-members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

Deep Kernel Learning
---------------------
.. autoclass:: atomai.models.dklGPR
    :members:
    :inherited-members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

Load trained models
--------------------
.. autofunction:: atomai.models.load_model
.. autofunction:: atomai.models.load_ensemble