Fine-tuning để phân loại đánh giá
Dự án của bạn tiếp tục và bây giờ bạn muốn phân loại các đánh giá thành tích cực, trung tính hoặc tiêu cực. Bạn sẽ dùng một mô hình đã fine-tune để tự động phân loại các đánh giá.
Các lớp TrainingArguments và Trainer đã được import từ transformers.
training_args, model, và tokenized_datasets cũng đã được import sẵn.
Bài tập này là một phần của khóa học
Reinforcement Learning from Human Feedback (RLHF)
Hướng dẫn bài tập
- Tạo các tập dữ liệu train và test bằng các tập con
"train"và"test"từ tậptokenized_datasets. - Khởi tạo lớp trainer.
- Thêm các tham số huấn luyện vào lớp.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# 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=____
)