Ajustarea fină pentru clasificarea recenziilor
Proiectul tău continuă, iar acum vrei să clasifici recenziile ca pozitive, neutre sau negative. Vei folosi un model ajustat fin pentru a categoriza automat recenziile.
Clasele TrainingArguments și Trainer au fost importate din transformers.
De asemenea, training_args, model și tokenized_datasets au fost pre-importate.
Acest exercițiu face parte din cursul
Reinforcement Learning from Human Feedback (RLHF)
Instrucțiuni pentru exercițiu
- Creează seturile de date de antrenament și de testare folosind subseturile
"train"și"test"din setul de datetokenized_datasets. - Inițializează clasa trainer.
- Adaugă argumentele de antrenament la clasă.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# 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=____
)