開始使用免費開始

為評論分類進行微調

你的專案持續進行中,你現在想要將評論分類為正向、中立或負向。你會使用一個已微調的模型,自動為評論進行分類。

TrainingArgumentsTrainer 類別已從 transformers 匯入。 training_argsmodeltokenized_datasets 也都已經預先匯入。

本練習屬於課程

Reinforcement Learning from Human Feedback(RLHF)

檢視課程

練習說明

  • 使用 tokenized_datasets 資料集中 "train""test" 子集合,建立訓練與測試資料集。
  • 初始化 trainer 類別。
  • 將訓練參數加入到該類別。

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

# 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=____
)
編輯並執行程式碼