การ Fine-tune สำหรับการจำแนกรีวิว
โปรเจกต์ดำเนินต่อไป และขั้นตอนถัดไปคือการจำแนกรีวิวออกเป็นเชิงบวก เชิงกลาง หรือเชิงลบ โดยใช้โมเดลที่ผ่านการ fine-tune เพื่อจัดหมวดหมู่รีวิวโดยอัตโนมัติ
คลาส TrainingArguments และ Trainer ถูก import มาจาก transformers แล้ว
รวมถึง training_args, model และ tokenized_datasets ก็ถูก pre-import ไว้เรียบร้อยแล้ว
แบบฝึกหัดนี้เป็นส่วนหนึ่งของหลักสูตร
Reinforcement Learning from Human Feedback (RLHF)
คำแนะนำการฝึกหัด
- สร้างชุดข้อมูล train และ test โดยใช้ subset
"train"และ"test"จากชุดข้อมูลtokenized_datasets - สร้าง instance ของคลาส trainer
- เพิ่ม training arguments ให้กับคลาส
แบบฝึกหัดเชิงโต้ตอบแบบลงมือทำ
ลองทำแบบฝึกหัดนี้โดยเติมโค้ดตัวอย่างนี้ให้สมบูรณ์
# 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=____
)