ComenzarEmpieza gratis

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)

Ver curso

Instrucciones del ejercicio

  • Crea los conjuntos de entrenamiento y prueba usando los subconjuntos "train" y "test" del conjunto de datos tokenized_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=____
)
Editar y ejecutar código