为 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}")