nlpatl.models.embeddings.sentence_transformers

class nlpatl.models.embeddings.sentence_transformers.SentenceTransformers(model_name_or_path, batch_size=16, name='sentence_transformers')[source]

Bases: nlpatl.models.embeddings.embeddings.Embeddings

A wrapper of transformers class.

Parameters

  • model_name_or_path (str) – sentence transformers model name.

  • batch_size (int) – Batch size of data processing. Default is 16

  • model_config (dict) – Model paramateters. Refer to https://www.sbert.net/docs/pretrained_models.html

  • name (str) – Name of this embeddings

>>> import nlpatl.models.embeddings as nme
>>> model = nme.SentenceTransformers()
convert(x)[source]
Parameters

x (np.ndarray) – Raw features

Returns

Vectors of features

Return type

np.ndarray