abbasgolestani/ag-nli-DeTS-sentence-similarity-v3-light
Este modelo fue entrenado usando la clase Cross-Encoder de SentenceTransformers. El modelo predice una puntuación entre 0 (no similar) y 1 (muy similar) para la similitud semántica de dos oraciones.
Como usar
Comparando 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-v3-light')
# 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-v3-light')
scores = model.predict([('Sentence 1', 'Sentence 2'), ('Sentence 3', 'Sentence 4')])
El modelo predecirá puntuaciones para los pares ('Sentence 1', 'Sentence 2') y ('Sentence 3', 'Sentence 4'). También puede usar este modelo sin SentenceTransformers y solo usando la clase AutoModel de Transformers.
Funcionalidades
- Clasificación de Texto
- Transformers
- PyTorch
- multi_nli
- electra
- extracción de características
- similitud de oraciones
Casos de uso
- Clasificación de oraciones según su similitud
- Evaluación de la similitud semántica entre pares de oraciones