Trainers and Predictors
========================

Single Model Trainers
----------------------

Base Trainer Class
^^^^^^^^^^^^^^^^^^^
.. autoclass:: atomai.trainers.BaseTrainer
    :members:
    :undoc-members:
    :member-order: bysource


Trainer for Semantic Segmentation
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: atomai.trainers.SegTrainer
    :members:
    :inherited-members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

ImSpec Trainer
^^^^^^^^^^^^^^^
.. autoclass:: atomai.trainers.ImSpecTrainer
    :members:
    :inherited-members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

Variational Inference Trainer
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: atomai.trainers.viBaseTrainer
    :members:
    :inherited-members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

Deep Ensemble Trainers
-----------------------

Ensemble Trainer
^^^^^^^^^^^^^^^^^
.. autoclass:: atomai.trainers.EnsembleTrainer
    :members:
    :inherited-members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

Single Model Predictors
------------------------

BasePredictor
^^^^^^^^^^^^^^
.. autoclass:: atomai.predictors.BasePredictor
    :members:
    :undoc-members:
    :member-order: bysource

SegPredictor
^^^^^^^^^^^^^
.. autoclass:: atomai.predictors.SegPredictor
    :members:
    :inherited-members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

ImSpecPredictor
^^^^^^^^^^^^^^^^
.. autoclass:: atomai.predictors.ImSpecPredictor
    :members:
    :inherited-members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

Deep Ensemble Predictors
-------------------------

Ensemble Predictor
^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: atomai.predictors.EnsemblePredictor
    :members:
    :undoc-members:
    :member-order: bysource
    :show-inheritance:

.. autofunction:: atomai.predictors.ensemble_locate