albert-small-kor-cross-encoder-v1
bongsoo
Clasificación de texto
Este modelo fue entrenado usando la clase SentenceTransformers Cross-Encoder. El modelo albert-small-kor-v1 se afinó como un cross-encoder.
Como usar
Los modelos preentrenados se pueden usar así:
from sentence_transformers import CrossEncoder
model = CrossEncoder('bongsoo/albert-small-kor-cross-encoder-v1')
scores = model.predict([('오늘 날씨가 좋다', '오늘 등산을 한다'), ('오늘 날씨가 흐리다', '오늘 비가 내린다')])
print(scores)
El modelo predecirá puntuaciones para los pares ('Frase 1', 'Frase 2') y ('Frase 3', 'Frase 4'). También se puede usar este modelo sin sentence_transformers utilizando la clase AutoModel de Transformers.
Funcionalidades
- Modelo ALBERT en coreano
- Utiliza transformadores y PyTorch
- Entrenado con la clase Cross-Encoder de SentenceTransformers
Casos de uso
- Clasificación de texto
- Emparejamiento de frases
- Predicción de similitud entre oraciones