开始使用免费开始使用

为 RLHF 对生成文本进行分类

现在,您希望对生成的评论进行分类。评估输出的一种方式是测量生成评论的正向程度。您可以使用分类器 lvwerra/distilbert-imdb 来完成此操作,也可以通过 Hugging Face 的 pipelines 来实例化它。

pipeline 库已从 transformers 预先导入。lvwerra/distilbert-imdb 模型已预加载为 model。分词器已预加载为 tokenizer

本练习是课程的一部分

来自人类反馈的强化学习(RLHF)

查看课程

练习说明

  • 使用 pipeline 函数,基于该模型创建一个情感分析(sentiment-analysis) pipeline。
  • 对给定的评论进行情感分类。

交互式实操练习

通过完成这段示例代码来试试这个练习。

# Create a sentiment analysis pipeline
sentiment_analyzer = pipeline(____, model=____, tokenizer=____)

review_text = "Surprisingly, the film is a very good one"

# Classify the sentiment of the review
sentiment = sentiment_analyzer(____)
print(f"Sentiment Analysis Result: {sentiment}")
编辑并运行代码