Kom igångKom igång gratis

Finjustering för recensionsklassificering

Projektet fortsätter och du vill nu klassificera recensionerna som positiva, neutrala eller negativa. Du använder en finjusterad modell för att kategorisera recensionerna automatiskt.

Klasserna TrainingArguments och Trainer har importerats från transformers. training_args, model och tokenized_datasets har också förimpporterats.

Den här övningen är en del av kursen

Reinforcement Learning from Human Feedback (RLHF)

Visa kurs

Övningsinstruktioner

  • Skapa tränings- och testdatamängderna med delmängderna "train" och "test" från datamängden tokenized_datasets.
  • Initiera tränarklassen.
  • Lägg till träningsargument i klassen.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

# 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=____
)
Redigera och kör kod