CommencerCommencer gratuitement

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)

Afficher le cours

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}")
Modifier et exécuter le code