pavlentiy/reviews-sentiment-multilingual-e5-base

pavlentiy
Similitud de oraciones

Este es un modelo de sentence-transformers que mapea oraciones y párrafos a un espacio vectorial denso de 768 dimensiones y puede ser usado para tareas como la agrupación o la búsqueda semántica.

Como usar

El uso de este modelo se facilita cuando tienes sentence-transformers instalado:

pip install -U sentence-transformers

Luego puedes usar el modelo de esta manera:

import numpy as np
from sentence_transformers import SentenceTransformer, util

sentences = ["Замечательный препарат, все пользуемся", "Последнее время данный препарат вызывает у меня сыпь"]
classes = ['негатив', 'нейтрально', 'позитив']

model = SentenceTransformer('pavlentiy/reviews-sentiment-multilingual-e5-base')
embeddings = model.encode(sentences)
embeddings_classes = model.encode(classes)

# Calcular similitudes coseno
cosine_scores = np.array(util.cos_sim(embeddings, embeddings_classes))

a = lambda t: {0:'негатив', 1:'нейтральная', 2:'позитив'}[t]
argmax = cosine_scores.argmax(axis=1)
result_classes = list(map(a, argmax))
print(result_classes)

Funcionalidades

Modelo de sentence-transformers
Mapeo de oraciones y párrafos a un espacio vectorial denso de 768 dimensiones
Utilizable para tareas de agrupación o búsqueda semántica

Casos de uso

Clasificación de sentimientos en múltiples idiomas
Agrupación de oraciones
Búsqueda semántica