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