為評論分類進行微調
你的專案持續進行中,你現在想要將評論分類為正向、中立或負向。你會使用一個已微調的模型,自動為評論進行分類。
TrainingArguments 與 Trainer 類別已從 transformers 匯入。
training_args、model 與 tokenized_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=____
)