Преобразование предложений в вектора с помощью Sentence Transformers в Python

Sentence Transformers - библиотека в Python, которая позволяет преобразовывать в вектора целые предложения/абзацы с целью их дальнейшей обработки (например кластеризации).

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

pip install -U sentence-transformers

или на конде:

conda install -c conda-forge sentence-transformers

Пример использования библиотеки:

Импортируем необходимый модуль

Загружаем нужную версию

Задаем предложения, которые нужно преобразовать в вектора

Непосредственно преобразовываем

from sentence_transformers import SentenceTransformer model = SentenceTransformer("all-MiniLM-L6-v2") sentences = ["The weather is lovely today.","It's so sunny outside!","He drove to the stadium.",] sentence_embeddings = model.encode(sentences) print(sentence_embeddings)

Больше полезной информации вы можете найти в моем Телеграм канале Бизнес, IT и статистика