要約の長さを調整する
pipeline()には、min_new_tokensとmax_new_tokensという2つのパラメータがあります。これらを使うと、生成される要約テキストの長さを短くしたり、長くしたり、あるいは特定の語数に調整できます。ストレージの制約がある場合や、読みやすくしたい場合、要約の質を改善したい場合に活用できます。
ここでは、この2つのパラメータを使って、短い要約と長い要約の両方を試してみましょう。
transformersライブラリのpipelineとoriginal_textはあらかじめ読み込まれています。
この演習はコースの一部です
Hugging Faceを使いこなす
実践的なインタラクティブ演習
このサンプルコードを完成させて、この演習に挑戦してみましょう。
# Generate a summary of original_text between 1 and 10 tokens
short_summarizer = pipeline(task="summarization", model="cnicu/t5-small-booksum", ____=1, ____=10)
short_summary_text = ____(original_text)
print(short_summary_text[0]["summary_text"])