Тонке налаштування для класифікації відгуків
Ваш проєкт триває, і тепер ви хочете класифікувати відгуки як позитивні, нейтральні або негативні. Ви використаєте модель із тонким налаштуванням, щоб автоматично категоризувати відгуки.
Класи TrainingArguments і Trainer було імпортовано з transformers.
Також попередньо імпортовано training_args, model та tokenized_datasets.
Ця вправа є частиною курсу
Reinforcement Learning from Human Feedback (RLHF)
Інструкції до вправи
- Створіть тренувальний і тестовий набори даних, використавши підвибірки
"train"і"test"з набору данихtokenized_datasets. - Ініціалізуйте клас тренера.
- Додайте параметри тренування до класу.
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
# 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=____
)