Zacznij terazZacznij za darmo

Dostrajanie modelu do klasyfikacji recenzji

Projekt trwa – teraz chcesz klasyfikować recenzje jako pozytywne, neutralne lub negatywne. Użyjesz dostrojonego modelu, aby automatycznie kategoryzować recenzje.

Klasy TrainingArguments i Trainer zostały już zaimportowane z transformers. Zmienne training_args, model oraz tokenized_datasets są również wstępnie zaimportowane.

To ćwiczenie jest częścią kursu

Uczenie przez wzmacnianie z informacją zwrotną od człowieka (RLHF)

Zobacz kurs

Instrukcje do ćwiczenia

  • Utwórz zbiory treningowy i testowy, używając podzbiorów "train" i "test" ze zbioru danych tokenized_datasets.
  • Zainicjalizuj klasę trenera.
  • Dodaj argumenty trenowania do klasy.

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

# 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=____
)
Edytuj i uruchom kod