Ajuste fino para clasificar reseñas
Tu proyecto continúa y ahora quieres clasificar las reseñas como positivas, neutras o negativas. Usarás un modelo ajustado finamente para categorizar las reseñas automáticamente.
Las clases TrainingArguments y Trainer se han importado desde transformers.
training_args, model y tokenized_datasets también se han preimportado.
Este ejercicio forma parte del curso
Reinforcement Learning from Human Feedback (RLHF)
Instrucciones del ejercicio
- Crea los conjuntos de entrenamiento y prueba usando los subconjuntos
"train"y"test"del conjunto de datostokenized_datasets. - Inicializa la clase del entrenador.
- Añade los argumentos de entrenamiento a la clase.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# Define the train and test datasets
training_dataset = ____
testing_dataset = ____
# Initialize the trainer class
trainer = ____(
# Add arguments to the class
model=model,
args=training_args,
train_dataset=____,
eval_dataset=____
)