class SentenceTransformerEncoder:
Sentence Transformers
.
References:
https://www.sbert.net/
def __init__(self, model_name: str = 'intfloat/e5-large-v2', **kwargs):
SentenceTransformerEmbedding
class
with the specified transformer model.
Parameters:
intfloat/e5-large-v2
) **kwargs (optional): Additional arguments of :class:SentenceTransformer
, such as :obj:prompts
etc.def embed_list(self, objs: list[str], **kwargs: Any):
def get_output_dim(self):