NASA-AIML/MIKA_Custom_IR
NASA-AIML
Similitud de oraciones
Modelo de transformadores de oraciones: mapea oraciones y párrafos a un espacio vectorial denso de 768 dimensiones y puede ser utilizado para tareas como agrupamiento o búsqueda semántica. El modelo está entrenado específicamente en documentos de ingeniería para la recuperación asimétrica de información. Su intención es identificar documentos de ingeniería relevantes a una consulta para su uso durante el diseño. Por ejemplo, un repositorio puede ser consultado para encontrar soporte para requisitos o aprender más sobre un tipo específico de falla.
Como usar
Usar este modelo se vuelve fácil cuando tienes instalado sentence-transformers:
pip install -U sentence-transformers
Luego puedes usar el modelo así:
from sentence_transformers import SentenceTransformer
sentences = ["Esta es una oración de ejemplo", "Cada oración es convertida"]
model = SentenceTransformer("NASA-AIML/MIKA_Custom_IR")
embeddings = model.encode(sentences)
print(embeddings)
Funcionalidades
- Modelado de similaridad de oraciones
- Transformadores de oraciones: sentence-transformers
- Utiliza PyTorch
- Modelo MPNet
- Extracción de características
- Compatible con AutoTrain
- Compatible con Puntos de Inferencia
- Región: EE.UU.
Casos de uso
- Agrupamiento de oraciones
- Búsqueda semántica
- Identificación de documentos de ingeniería relevantes para una consulta
- Soporte en la búsqueda de requisitos
- Aprendizaje sobre tipos específicos de fallas en diseño