标签:

• 摘要生成 许可证: mit

BART（大型模型），基于自定义合同数据集微调

BART模型在英语语言上预训练，并针对服务条款的抽象摘要任务进行了微调

模型描述

BART是一个采用编码器-编码器架构（序列到序列）的Transformer模型，包含双向（类似BERT）的编码器和自回归（类似GPT）的解码器。BART通过以下方式预训练：(1) 使用任意噪声函数破坏文本，(2) 学习模型以重建原始文本。

BART在文本生成任务（如摘要、翻译）上经过微调后表现尤为出色，同时在理解任务（如文本分类、问答）上也表现良好。此特定检查点已在CNN每日邮报（大量文本-摘要对集合）上进行了微调。

预期用途与限制

您可将此模型用于文本摘要任务。

使用方法

以下是使用pipeline API调用该模型的方式：

from transformers import pipeline

summarizer = pipeline("summarization", model="ML-unipi/bart-large-tos")

ARTICLE = """ 纽约（CNN）当Liana Barrientos 23岁时，她在纽约威彻斯特郡结婚。
一年后，她再次在威彻斯特郡结婚，但对象是另一个男人且未与首任丈夫离婚。
那次婚姻仅18天后，她又一次结婚。随后，Barrientos又说了五次"我愿意"，有时两次婚礼间隔仅两周。
2010年，她再次结婚，这次是在布朗克斯区。在结婚许可证申请中，她声称这是其"第一次也是唯一一次"婚姻。
根据法庭文件，现年39岁的Barrientos面临两项"一级提供虚假文书"刑事指控，涉及她在2010年结婚申请中的虚假陈述。
检方称这些婚姻是移民骗局的一部分。
据其律师Christopher Wright表示，周五她在布朗克斯最高法院拒不认罪，律师拒绝进一步评论。
警方发言人Annette Markowski侦探称，离开法庭后，Barrientos因涉嫌通过紧急出口潜入纽约地铁而被捕，并被指控盗窃服务和刑事侵入。Barrientos总计结婚10次，其中9次发生在1999至2002年间。
所有婚姻要么发生在威彻斯特郡、长岛、新泽西州，要么在布朗克斯区。检方表示，据信她仍与四名男子保持婚姻关系，且曾一度同时与八名男子结婚。
检方称移民骗局涉及部分丈夫，他们在婚后不久就申请了永久居留身份。
离婚仅在这些申请获批后才发生。目前尚不清楚是否有男子会被起诉。
该案件由移民海关执法局和国土安全部调查分局移交布朗克斯地区检察官办公室处理。其中七名男子来自所谓"红色标记"国家，包括埃及、土耳其、格鲁吉亚、巴基斯坦和马里。
她的第八任丈夫Rashid Rajput在联合反恐任务组调查后，于2006年被驱逐回原籍巴基斯坦。
若定罪，Barrientos将面临最高四年监禁。其下次出庭时间定于5月18日。
"""
print(summarizer(ARTICLE, max_length=130, min_length=30, do_sample=False))
>>> [{'summary_text': '39岁的Liana Barrientos被指控两项"一级提供虚假文书"罪名。她总计结婚10次，其中9次发生在1999至2002年间。据信目前仍与四名男子保持婚姻关系。'}]