abbasgolestani/ag-nli-DeTS-sentence-similarity-v4
abbasgolestani
Clasificación de texto
Este modelo se entrenó utilizando la clase Cross-Encoder de SentenceTransformers. Fue entrenado en 6 diferentes conjuntos de datos nli. El modelo predecirá una puntuación entre 0 (no similar) y 1 (muy similar) para la similitud semántica de dos oraciones.
Como usar
from sentence_transformers import CrossEncoder
model = CrossEncoder('abbasgolestani/ag-nli-DeTS-sentence-similarity-v4')
# 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("{} {} Score: {:.4f}".format(sentences1[i], sentences2[i], scores1[i]))
# Uso de modelos preentrenados
scores = model.predict([('Sentence 1', 'Sentence 2'), ('Sentence 3', 'Sentence 4')])
El modelo predecirá puntajes para los pares ('Sentence 1', 'Sentence 2') y ('Sentence 3', 'Sentence 4').
Puedes usar este modelo también sin sentence_transformers y simplemente usando 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
- Comparar la similitud de oraciones para tareas de procesamiento de lenguaje natural.
- Evaluar la coherencia semántica en textos.
- Desarrollar aplicaciones que requieran análisis de similitud textual.