navteca/ms-marco-MiniLM-L-6-v2
navteca
Clasificación de texto
El modelo navteca/ms-marco-MiniLM-L-6-v2 puede ser utilizado para la Recuperación de Información: Dada una consulta, codifica la consulta con todos los pasajes posibles (por ejemplo, recuperados con ElasticSearch). Luego, ordena los pasajes en orden decreciente. Consulta SBERT.net Retrieve & Re-rank para más detalles. El código de entrenamiento está disponible aquí: SBERT.net Training MS Marco
Como usar
El uso se facilita cuando tienes SentenceTransformers instalado. Luego, puedes usar los modelos preentrenados así:
from sentence_transformers import CrossEncoder
model = CrossEncoder('model_name', max_length=512)
scores = model.predict([('Query', 'Paragraph1'), ('Query', 'Paragraph2')])
Funcionalidades
- Modelo para Recuperación de Información
- Preentrenado para la tarea de MS Marco Passage Ranking
- Basado en SentenceTransformers y PyTorch
- Compatible con JAX
- Licenciado bajo MIT
Casos de uso
- Recuperación de Información
- Clasificación de texto
- Reordenamiento de pasajes