Training
===============

TOC
-----------

.. toctree::
   :maxdepth: 2

   src.training.losses
   src.training.subclasses
   src.training.trainers


Base
------------------------

.. automodule:: src.training.base
   :members:
   :undoc-members:
   :show-inheritance:

Multilabel BERT
------------------------------------

.. automodule:: src.training.bert_multilabel
   :members:
   :undoc-members:
   :show-inheritance:

Multilabel Distilbert
------------------------------

.. automodule:: src.training.distilbert
   :members:
   :undoc-members:
   :show-inheritance:

Setfit
--------------------------

.. automodule:: src.training.setfit
   :members:
   :undoc-members:
   :show-inheritance:

Other
---------------

.. automodule:: src.training
   :members:
   :undoc-members:
   :show-inheritance: