ComenzarEmpieza gratis

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.

Este ejercicio forma parte del curso

Reinforcement Learning from Human Feedback (RLHF)

Ver curso

Instrucciones del ejercicio

  • Use the pipeline function to create a sentiment-analysis pipeline with the model.
  • Classify the sentiment of the review provided.

Ejercicio interactivo práctico

Prueba este ejercicio y completa el código de muestra.

# 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}")
Editar y ejecutar código