Back to Sentence Transformers

Modules

docs/package_reference/sentence_transformer/modules.md

5.4.11.2 KB
Original Source

Modules

sentence_transformers.sentence_transformer.modules defines different building blocks, a.k.a. Modules, that can be used to create SentenceTransformer models from scratch. For more details, see Creating Custom Models.

See also the modules from sentence_transformers.base.modules in Base > Modules.

Main Modules

{eval-rst}
.. autoclass:: sentence_transformers.sentence_transformer.modules.Pooling
.. autoclass:: sentence_transformers.sentence_transformer.modules.Normalize
.. autoclass:: sentence_transformers.sentence_transformer.modules.StaticEmbedding
    :members: from_model2vec, from_distillation

Further Modules

{eval-rst}
.. autoclass:: sentence_transformers.sentence_transformer.modules.BoW
.. autoclass:: sentence_transformers.sentence_transformer.modules.CNN
.. autoclass:: sentence_transformers.sentence_transformer.modules.LSTM
.. autoclass:: sentence_transformers.sentence_transformer.modules.WeightedLayerPooling
.. autoclass:: sentence_transformers.sentence_transformer.modules.WordEmbeddings
.. autoclass:: sentence_transformers.sentence_transformer.modules.WordWeights