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
<Kurs>Reinforcement Learning aus menschlichem Feedback (RLHF)</Kurs>Übungsanweisungen
- 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 praktische Übung
Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.
# 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=____
)