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)
Oefeninstructies
- Maak de train- en test-datasets aan met de subsets
"train"en"test"uit de gegevenssettokenized_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=____
)