Fine-tuning untuk klasifikasi ulasan
Proyek Anda berlanjut dan sekarang Anda ingin mengklasifikasikan ulasan sebagai positif, netral, atau negatif. Anda akan menggunakan model yang telah di-fine-tune untuk mengategorikan ulasan secara otomatis.
Kelas TrainingArguments dan Trainer telah diimpor dari transformers.
training_args, model, dan tokenized_datasets juga telah diimpor sebelumnya.
Latihan ini adalah bagian dari kursus
Reinforcement Learning from Human Feedback (RLHF)
Petunjuk latihan
- Buat himpunan data train dan test menggunakan subhimpunan
"train"dan"test"dari himpunan datatokenized_datasets. - Inisialisasi kelas trainer.
- Tambahkan argumen pelatihan ke kelas.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# 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=____
)