Neural Nets
===========


Fully convolutional neural networks
-----------------------------------
.. autoclass:: atomai.nets.Unet
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

.. autoclass:: atomai.nets.dilnet
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

.. autoclass:: atomai.nets.ResHedNet
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

.. autoclass:: atomai.nets.SegResNet
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

Neural Networks for VAE
------------------------
.. autoclass:: atomai.nets.fcEncoderNet
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

.. autoclass:: atomai.nets.convEncoderNet
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

.. autoclass:: atomai.nets.fcDecoderNet
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

.. autoclass:: atomai.nets.convDecoderNet
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

.. autoclass:: atomai.nets.rDecoderNet
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

Neural Networks for ImSpec
---------------------------
.. autoclass:: atomai.nets.SignalEncoder
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

.. autoclass:: atomai.nets.SignalDecoder
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

.. autoclass:: atomai.nets.SignalED
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:


Building blocks
----------------
.. autoclass:: atomai.nets.ConvBlock
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

.. autoclass:: atomai.nets.DilatedBlock
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

.. autoclass:: atomai.nets.UpsampleBlock
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

.. autoclass:: atomai.nets.ResBlock
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

.. autoclass:: atomai.nets.ResModule
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance: