abbasgolestani/ag-nli-DeTS-sentence-similarity-v1
Este modelo fue entrenado utilizando la clase Cross-Encoder de SentenceTransformers para similitud de oraciones. Fue entrenado en 6 datasets diferentes de NLI y predecirá una puntuación entre 0 (no similar) y 1 (muy similar) para la similitud semántica de dos oraciones.
Como usar
Comparar cada oración de la matriz sentences1 con la oración correspondiente de la matriz sentences2, como comparar la primera oración de cada matriz, luego comparar la segunda oración de cada matriz, etc.
from sentence_transformers import CrossEncoder
model = CrossEncoder('abbasgolestani/ag-nli-DeTS-sentence-similarity-v1')
# Dos listas de oraciones
sentences1 = ['I am honored to be given the opportunity to help make our company better',
'I love my job and what I do here',
'I am excited about our company’s vision']
sentences2 = ['I am hopeful about the future of our company',
'My work is aligning with my passion',
'Definitely our company vision will be the next breakthrough to change the world and I’m so happy and proud to work here']
pairs = zip(sentences1, sentences2)
list_pairs = list(pairs)
scores1 = model.predict(list_pairs, show_progress_bar=False)
print(scores1)
for i in range(len(sentences1)):
print("{} \t\t {} \t\t Score: {:.4f}".format(sentences1[i], sentences2[i], scores1[i]))
Modelos preentrenados se pueden usar así:
from sentence_transformers import CrossEncoder
model = CrossEncoder('abbasgolestani/ag-nli-DeTS-sentence-similarity-v1')
scores = model.predict([('Sentence 1', 'Sentence 2'), ('Sentence 3', 'Sentence 4')])
print(scores)
El modelo predecirá puntuaciones para los pares ('Sentence 1', 'Sentence 2') y ('Sentence 3', 'Sentence 4'). También puedes usar este modelo sin sentence_transformers y utilizando solo la clase Transformers AutoModel.
Funcionalidades
- Clasificación de texto
- Transformers
- PyTorch
- Extracción de características
- Similitud de oraciones
- Compatible con AutoTrain
- Compatible con Endpoints
Casos de uso
- Clasificación de texto
- Medición de la similitud de oraciones