abbasgolestani/ag-nli-DeTS-sentence-similarity-v2

abbasgolestani
Clasificación de texto

Este modelo fue entrenado usando la clase Cross-Encoder de SentenceTransformers. Fue entrenado en 6 diferentes conjuntos de datos de NLI. El modelo predice 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 del array sentences1 con la oración correspondiente del array sentences2, como comparando la primera oración de cada array, luego comparando la segunda oración de cada array, etc.

from sentence_transformers import CrossEncoder

model = CrossEncoder('abbasgolestani/ag-nli-DeTS-sentence-similarity-v2')

# Dos listas de oraciones
sentences1 = ['Me siento honrado de tener la oportunidad de ayudar a mejorar nuestra empresa',
'Amo mi trabajo y lo que hago aquí',
'Estoy emocionado con la visión de nuestra empresa']

sentences2 = ['Tengo esperanzas sobre el futuro de nuestra empresa',
'Mi trabajo se está alineando con mi pasión',
'Sin duda, la visión de nuestra empresa será el próximo gran avance para cambiar el mundo y estoy muy feliz y orgulloso de trabajar aquí']

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 Puntaje: {:.4f}".format(sentences1[i], sentences2[i], scores1[i]))

Modelos preentrenados pueden ser utilizados de esta manera:

from sentence_transformers import CrossEncoder
model = CrossEncoder('abbasgolestani/ag-nli-DeTS-sentence-similarity-v2')

scores = model.predict([('Oración 1', 'Oración 2'), ('Oración 3', 'Oración 4')])

El modelo predecirá puntajes para los pares ('Oración 1', 'Oración 2') y ('Oración 3', 'Oración 4').

Puedes usar este modelo también sin SentenceTransformers y solo usando la clase AutoModel de Transformers.

Funcionalidades

Clasificación de texto
Transformers
PyTorch
Roberta
Extracción de características
Similitud de oraciones
Compatible con AutoTrain
Compatible con Endpoints

Casos de uso

Medir la similitud semántica entre dos oraciones
Evaluar la cohesión de contenido en análisis textual
Filtrar resultados de búsqueda para mostrar aquellos contextualmente más relevantes

Recibe las últimas noticias y actualizaciones sobre el mundo de IA directamente en tu bandeja de entrada.