调整摘要长度
pipeline() 函数有两个重要参数:min_new_tokens 和 max_new_tokens。它们可用于调整生成摘要文本的长度,使其更短、更长,或控制在一定的词数范围内。如果存在空间限制(例如存储较小)、需要提升可读性,或希望改进摘要质量,您都可能需要这样做。
您将通过设置这两个参数为较小范围,然后再设置为更大范围,来分别体验短摘要与长摘要。
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"])