Deehan1866/finetuned-roberta-large
Deehan1866
Similitud de oraciones
SentenceTransformer basado en FacebookAI/roberta-large. Este es un modelo de transformers de oraciones afinado a partir del modelo FacebookAI/roberta-large en el conjunto de datos PiC/phrase_similarity. Mapea oraciones y párrafos a un espacio vectorial denso de 1024 dimensiones y puede ser utilizado para similitud textual semántica, búsqueda semántica, minería de paráfrasis, clasificación de texto, agrupamiento y más.
Como usar
Uso Directo (Sentence Transformers)
# Primero instala la biblioteca Sentence Transformers:
pip install -U sentence-transformers
# Luego, puedes cargar este modelo y ejecutar inferencias.
from sentence_transformers import SentenceTransformer
# Descargar del Hub de 🤗
model = SentenceTransformer("Deehan1866/finetuned-roberta-large")
# Ejecutar inferencias
sentences = [
"Ella quiere escribir sobre Keima pero sufre un caso grave de bloqueo del escritor.",
"Ella quiere escribir sobre Keima pero sufre un gran caso de bloqueo del escritor.",
'especifico estado médico del movimiento y la condición general del movimiento son condiciones bajo las cuales las contradicciones pueden moverse.',
]
embeddings = model.encode(sentences)
print(embeddings.shape)
# [3, 1024]
# Obtener las puntuaciones de similitud para las incrustaciones
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [3, 3]
Funcionalidades
- Modelo: Sentence Transformer
- Modelo base: FacebookAI/roberta-large
- Longitud máxima de secuencia: 512 tokens
- Dimensionalidad de salida: 1024 tokens
- Función de similitud: Similitud del coseno
- Conjunto de datos de entrenamiento: PiC/phrase_similarity
Casos de uso
- Similitud textual semántica
- Búsqueda semántica
- Minería de paráfrasis
- Clasificación de texto
- Agrupamiento de texto