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)
Övningsinstruktioner
- Skapa tränings- och testdatamängderna med delmängderna
"train"och"test"från datamängdentokenized_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=____
)