IniziaInizia gratis

Fine-tuning per la classificazione delle recensioni

Il tuo progetto continua e ora vuoi classificare le recensioni come positive, neutre o negative. Userai un modello messo a punto (fine-tuned) per categorizzare automaticamente le recensioni.

Le classi TrainingArguments e Trainer sono state importate da transformers. Sono già stati pre-importati anche training_args, model e tokenized_datasets.

Questo esercizio fa parte del corso

Reinforcement Learning from Human Feedback (RLHF)

Visualizza il corso

Istruzioni dell'esercizio

  • Crea i dataset di train e test usando i sottoinsiemi "train" e "test" del dataset tokenized_datasets.
  • Inizializza la classe del trainer.
  • Aggiungi gli argomenti di training alla classe.

Esercizio pratico interattivo

Prova a risolvere questo esercizio completando il codice di esempio.

# 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=____
)
Modifica ed esegui il codice