ÎncepețiÎncepe gratuit

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)

Vezi cursul

Instrucțiuni pentru exercițiu

  • Creează seturile de date de antrenament și de testare folosind subseturile "train" și "test" din setul de date tokenized_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=____
)
Editează și rulează codul