This notebook regroups the code sample of the video below, which is a part of the Hugging Face course.

In [ ]:

#@title
from IPython.display import HTML

HTML('<iframe width="560" height="315" src="https://www.youtube.com/embed/TMshhnrEXlg?rel=0&amp;controls=0&amp;showinfo=0" frameborder="0" allowfullscreen></iframe>')

Out[ ]:

Install the Transformers and Datasets libraries to run this notebook.

In [ ]:

! pip install datasets transformers[sentencepiece]

In [ ]:

! pip install nltk rouge_score

In [ ]:

from datasets import load_metric

rouge = load_metric("rouge")
predictions = ["I really loved reading the Hunger Games"]
references = ["I loved reading the Hunger Games"]
rouge.compute(predictions=predictions, references=references)

In [ ]: