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)
Instrukcje do ćwiczenia
- Utwórz zbiory treningowy i testowy, używając podzbiorów
"train"i"test"ze zbioru danychtokenized_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=____
)