MulaiMulai sekarang secara gratis

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)

Lihat Kursus

Petunjuk latihan

  • Buat himpunan data train dan test menggunakan subhimpunan "train" dan "test" dari himpunan data tokenized_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=____
)
Edit dan Jalankan Kode