Feinabstimmung für die Review-Klassifikation
Dein Projekt geht weiter, und du möchtest die Reviews nun als positiv, neutral oder negativ klassifizieren. Du verwendest ein feinabgestimmtes Modell, um die Reviews automatisch zu kategorisieren.
Die Klassen TrainingArguments und Trainer wurden aus transformers importiert.
Die Objekte training_args, model und tokenized_datasets wurden ebenfalls bereits importiert.
Diese Übung ist Teil des Kurses
Reinforcement Learning aus menschlichem Feedback (RLHF)
Anleitung zur Übung
- Erstelle die Trainings- und Test-Datasets, indem du die Teilmengen
"train"und"test"aus dem Datasettokenized_datasetsverwendest. - Initialisiere die Trainer-Klasse.
- Füge der Klasse die Trainingsargumente hinzu.
Interaktive Übung
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
# 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=____
)