用于评论分类的微调
项目继续推进,您现在希望将评论分类为正面、中性或负面。您将使用一个经过微调的模型来自动归类这些评论。
TrainingArguments 和 Trainer 类已从 transformers 导入。
training_args、model 和 tokenized_datasets 也已预先导入。
本练习是课程的一部分
来自人类反馈的强化学习(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=____
)