Aan de slagGa gratis aan de slag

Fine-tunen voor reviewclassificatie

Je project gaat verder en je wilt de reviews nu classificeren als positief, neutraal of negatief. Je gebruikt een gefine-tuned model om de reviews automatisch te categoriseren.

De klassen TrainingArguments en Trainer zijn geïmporteerd uit transformers. De training_args, model en tokenized_datasets zijn ook al vooraf geïmporteerd.

Deze oefening maakt deel uit van de cursus

Reinforcement Learning from Human Feedback (RLHF)

Cursus bekijken

Oefeninstructies

  • Maak de train- en test-datasets aan met de subsets "train" en "test" uit de gegevensset tokenized_datasets.
  • Initialiseer de Trainer-klasse.
  • Voeg trainingsargumenten toe aan de klasse.

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

# 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=____
)
Code bewerken en uitvoeren