nasa-ibm-st.38m
nasa-impact
Similitud de oraciones
nasa-smd-ibm-st.38m es un modelo transformador de oraciones Bi-encoder, ajustado a partir de una versión destilada del modelo codificador nasa-smd-ibm-v0.1. Es una versión más pequeña de nasa-smd-ibm-st con mejor rendimiento, utilizando menos parámetros. Está entrenado con 362 millones de ejemplos junto con un conjunto de datos específico de dominio de 2.6 millones de ejemplos de documentos curados por la Dirección de la Misión Científica de la NASA (SMD). Con este modelo, buscamos mejorar las tecnologías de lenguaje natural como la recuperación de información y la búsqueda inteligente aplicada a las aplicaciones de NLP de la SMD.
Como usar
from sentence_transformers import SentenceTransformer, util
model = SentenceTransformer("nasa-impact/nasa-smd-ibm-st.38m")
input_queries = [
'query: how much protein should a female eat', 'query: summit define']
input_passages = [
"As a general guideline, the CDC's average requirement of protein for women ages 19 to 70 is 46 grams per day.\nBut, as you can see from this chart, you'll need to increase that if you're expecting or training for a marathon. Check out the chart below to see how much protein you should be eating each day.",
"Definition of summit for English Language Learners. : 1 the highest point of a mountain : the top of a mountain. : 2 the highest level. : 3 a meeting or series of meetings between the leaders of two or more governments."]
query_embeddings = model.encode(input_queries)
passage_embeddings = model.encode(input_passages)
print(util.cos_sim(query_embeddings, passage_embeddings))
Funcionalidades
- Modelo transformador de oraciones Bi-encoder
- Ajustado desde la versión destilada del modelo codificador nasa-smd-ibm-v0.1
- 362 millones de ejemplos de entrenamiento más 2.6 millones de ejemplos específicos del dominio
- Optimizador de relevancia de similitud de coseno
- Compatible con PyTorch 1.9.1 y sentence-transformers 4.30.2
Casos de uso
- Recuperación de Información
- Búsqueda de Similitud de Oraciones
- Casos de uso científicos relacionados con la SMD de la NASA