开始使用免费开始使用

用于评论分类的微调

项目继续推进,您现在希望将评论分类为正面、中性或负面。您将使用一个经过微调的模型来自动归类这些评论。

TrainingArgumentsTrainer 类已从 transformers 导入。 training_argsmodeltokenized_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=____
)
编辑并运行代码