API References
~~~~~~~~~~~~~~

Similarity Model
----------------

.. py:currentmodule:: quaterion_models.model

.. autosummary::
    :nosignatures:

    SimilarityModel

Encoders
--------

.. py:currentmodule:: quaterion_models.encoders

.. autosummary::
    :nosignatures:

    ~encoder.Encoder
    ~switch_encoder.SwitchEncoder
    ~extras.fasttext_encoder.FasttextEncoder


Head Layers
-----------

.. py:currentmodule:: quaterion_models.heads

.. autosummary::
    :nosignatures:

    ~empty_head.EmptyHead
    ~gated_head.GatedHead
    ~sequential_head.SequentialHead
    ~skip_connection_head.SkipConnectionHead
    ~softmax_head.SoftmaxEmbeddingsHead
    ~stacked_projection_head.StackedProjectionHead
    ~widening_head.WideningHead