譯者 | 布加迪

審校 | 重樓

若要使用Hugging Face的BART模型總結(jié)文本，請加載模型和分詞器，輸入文本，隨后模型就會生成簡潔的摘要。

BART是一個幫助總結(jié)文本的工具。它可以使長文章變得更短，更容易閱讀。這有助于你快速找到要點。BART的工作原理是，分析整段文本以理解其上下文。然后，它通過保留重要的部分并刪除不太重要的部分來生成摘要。

有了BART，你可以總結(jié)文章、報告及其他文本。它側(cè)重于關(guān)鍵信息，以創(chuàng)建清晰簡潔的版本。Hugging Face Transformers是一個庫，讓使用BART變得簡單。我們在本文中將介紹如何設(shè)置BART和創(chuàng)建摘要。

為什么使用BART總結(jié)文本？

BART對于文本總結(jié)非常有效，因為它可以：

• 理解上下文：BART可以很好地閱讀和理解長文本。它找到要點，做好總結(jié)。
• 生成連貫的摘要：BART生成易于閱讀的摘要。它保留了重要的細節(jié)，刪除了不需要的信息。
• 處理各種類型的文本：BART可以總結(jié)多種類型的文本，比如新聞文章、研究論文或故事。它很靈活，可以很好地處理不同的內(nèi)容。

現(xiàn)在不妨看看如何使用BART模型和Hugging Face Transformers來總結(jié)文本。

搭建環(huán)境

在使用BART模型之前，確保已安裝了必要的庫。你將需要Hugging Face Transformers庫。

pip install transformers

加載BART模型

接下來，你需要搭建摘要管道。你可以使用以下代碼加載預(yù)訓(xùn)練的BART模型：

from transformers import pipeline
# Load the summarization pipeline with the BART model
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")

• summarizer：存儲摘要管道的變量。
• pipeline：一個由Hugging Face提供的高級API，便于訪問各種模型。
• summarization：指定要執(zhí)行的任務(wù)，即文本摘要。
• model="facebook/ BART -large-cnn"：加載BART模型，該模型為總結(jié)文本而預(yù)先訓(xùn)練。

準備輸入文本

接下來，你需要準備想要總結(jié)的輸入文本。輸入文本需要分成更小的部分（名為詞元）。

input_text = """
Climate change means a long-term change in temperature and weather. It can happen in one place or the whole Earth. Right now, climate change is happening in many areas. It affects nature, water, food, and health. Scientists see changes in the climate over time. Most of these changes are caused by human actions. Activities like burning fossil fuels and cutting down trees lead to climate change. These actions increase greenhouse gases in the air. Greenhouse gases hold heat in the air and make the Earth hotter. This causes global temperatures to rise.
"""

總結(jié)文本

要總結(jié)文本，只需將input_text傳遞給summarizer管道。

# Generate the summary
summary = summarizer(input_text, max_length=50, min_length=25, do_sample=False)

# Output the summarized text
print(summary[0]['summary_text'])

• max_length：以詞元的形式定義生成的總結(jié)的最大長度。
• min_length：設(shè)置總結(jié)的最小長度。這確保總結(jié)不會太簡短。
• do_sample=False：通過使用貪婪解碼而不是采樣，確保確定性結(jié)果。

這將打印輸出輸入文本的較短版本。

Climate change means a long-term change in temperature and weather. Activities like burning fossil fuels and cutting down trees lead to climate change. Greenhouse gases hold heat in the air and make the Earth hotter.

結(jié)論

使用BART模型和Hugging Face Transformers是一種總結(jié)文本的簡潔方法。你可以快速設(shè)置它，并開始總結(jié)，只需幾個簡單的步驟。首先，加載預(yù)訓(xùn)練的模型和分詞器，然后輸入文本，模型將制作更簡短的版本。這可以節(jié)省時間，并幫助你查看重要的細節(jié)。現(xiàn)在就開始使用BART，讓文本總結(jié)簡單又快速！

原文標題：How to Summarize Texts Using the BART Model with Hugging Face Transformers，作者：Jayita Gulati