Classifying generated text for RLHF
You now want to categorize the generated reviews. One of the ways you can evaluate the output is by measuring the positivity of the generated reviews using the classifier lvwerra/distilbert-imdb
, which you can also instantiate using Hugging Face pipelines.
The pipeline
library has been pre-imported from transformers
. The lvwerra/distilbert-imdb
model has been pre-loaded as model
. The tokenizer has been pre-loaded as tokenizer
.
Cet exercice fait partie du cours
Reinforcement Learning from Human Feedback (RLHF)
Instructions
- Use the
pipeline
function to create a sentiment-analysis pipeline with the model. - Classify the sentiment of the review provided.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# 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}")