Začněte nyníZačněte zdarma

Fine-tuning pro klasifikaci recenzí

Projekt pokračuje a teď chceš klasifikovat recenze jako pozitivní, neutrální nebo negativní. K automatickému kategorizování recenzí použiješ fine-tuned model.

Třídy TrainingArguments a Trainer jsou již naimportované z transformers. Proměnné training_args, model a tokenized_datasets jsou také předem naimportované.

Toto cvičení je součástí kurzu

Reinforcement Learning from Human Feedback (RLHF)

Zobrazit kurz

Pokyny k cvičení

  • Vytvoř trénovací a testovací datové sady pomocí podmnožin "train" a "test" z datasetu tokenized_datasets.
  • Inicializuj třídu tréneru.
  • Přidej do třídy argumenty trénování.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

# 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=____
)
Upravit a spustit kód