Как добиться встраивания текста с помощью BERT?

Question

Я пытаюсь создать функцию встраивания текста с помощью BERT. Там говорилось, что BERT может встраивать текст. Однако я не могу найти функцию внедрения в учебнике BERT. Вот ссылка, которую я нашел: https://huggingface.co/docs/transformers/model_doc/bert Кто-нибудь знает ресурс для встраивания текста BERT? Или есть другие имена, которые представляют это? Я знаю, что для OpenAI есть функция, просто вызывающая OpenAIEmbeddings(). Я просто хочу найти подобную функцию. Спасибо!

Я просто хочу найти аналогичную функцию, например OpenAIEmbeddings().

Answer 1

Можете ли вы попробовать следующее:

Установите необходимый пакет:

      pip install -U sentence-transformers

КОД:

      from sentence_transformers import SentenceTransformer
model = SentenceTransformer('paraphrase-MiniLM-L6-v2')

# Sentences we want to encode. Example:
sentence = ['This framework generates embeddings for each input sentence']

# Sentences are encoded by calling model.encode()
embedding = model.encode(sentence)