Aprendizaje contrastivo + Prompt + FGSM para lograr la clasificación del texto

Este artículo se basa principalmente en la tarea de clasificación de texto de Robert. Sobre esta base, consideramos la fusión del aprendizaje contrastivo, el entrenamiento rápido y de confrontación para mejorar la capacidad de clasificación de texto del modelo. Tengo train.txt, test.txt, dev.txt tres archivos, cada archivo contiene dos columnas de contenido de texto y etiquetas, según la implementación de pytorch.

Tabla de contenido

1. Cargar bibliotecas relacionadas

2. Cargue el conjunto de datos

3. Aumento de datos para el aprendizaje contrastivo

Supongo que te gusta

Origin blog.csdn.net/weixin_43734080/article/details/131139455
Recomendado
Clasificación