Дообучение для классификации отзывов
Работа над проектом продолжается: теперь вам нужно классифицировать отзывы как положительные, нейтральные или отрицательные. Для автоматической категоризации отзывов вы воспользуетесь дообученной моделью.
Классы TrainingArguments и Trainer уже импортированы из transformers.
Также предварительно импортированы training_args, model и tokenized_datasets.
Это упражнение является частью курса
Обучение с подкреплением на основе обратной связи от людей (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=____
)