模型:

mrm8488/camembert2camembert_shared-finetuned-french-summarization

英文

法语RoBERTa2RoBERTa(共享)在MLSUM FR上进行细调的摘要

模型

camembert-base (RoBERTa检查点)

数据集

MLSUM是第一个大规模的多语言摘要数据集。该数据集从在线新闻中获得,包含五种不同语言(即法语、德语、西班牙语、俄语、土耳其语)的150万+文章/摘要对。与来自流行的CNN/Daily Mail数据集中的英语报纸一起,这些收集到的数据形成了一个大规模的多语言数据集,可以为文本摘要研究社区提供新的研究方向。我们报告了基于最先进系统的跨语言比较分析。这些分析突出了现有偏见,从而促使使用多语言数据集。 MLSUM fr

结果

Set Metric # Score
Test Rouge2 - mid -precision 14.47
Test Rouge2 - mid - recall 12.90
Test Rouge2 - mid - fmeasure 13.30

用法

import torch
from transformers import RobertaTokenizerFast, EncoderDecoderModel
device = 'cuda' if torch.cuda.is_available() else 'cpu'
ckpt = 'mrm8488/camembert2camembert_shared-finetuned-french-summarization'
tokenizer = RobertaTokenizerFast.from_pretrained(ckpt)
model = EncoderDecoderModel.from_pretrained(ckpt).to(device)
def generate_summary(text):
   inputs = tokenizer([text], padding="max_length", truncation=True, max_length=512, return_tensors="pt")
   input_ids = inputs.input_ids.to(device)
   attention_mask = inputs.attention_mask.to(device)
   output = model.generate(input_ids, attention_mask=attention_mask)
   return tokenizer.decode(output[0], skip_special_tokens=True)
   
text = "Un nuage de fumée juste après l’explosion, le 1er juin 2019. Une déflagration dans une importante usine d’explosifs du centre de la Russie a fait au moins 79 blessés samedi 1er juin. L’explosion a eu lieu dans l’usine Kristall à Dzerzhinsk, une ville située à environ 400 kilomètres à l’est de Moscou, dans la région de Nijni-Novgorod. « Il y a eu une explosion technique dans l’un des ateliers, suivie d’un incendie qui s’est propagé sur une centaine de mètres carrés », a expliqué un porte-parole des services d’urgence. Des images circulant sur les réseaux sociaux montraient un énorme nuage de fumée après l’explosion. Cinq bâtiments de l’usine et près de 180 bâtiments résidentiels ont été endommagés par l’explosion, selon les autorités municipales. Une enquête pour de potentielles violations des normes de sécurité a été ouverte. Fragments de shrapnel Les blessés ont été soignés après avoir été atteints par des fragments issus de l’explosion, a précisé une porte-parole des autorités sanitaires citée par Interfax. « Nous parlons de blessures par shrapnel d’une gravité moyenne et modérée », a-t-elle précisé. Selon des représentants de Kristall, cinq personnes travaillaient dans la zone où s’est produite l’explosion. Elles ont pu être évacuées en sécurité. Les pompiers locaux ont rapporté n’avoir aucune information sur des personnes qui se trouveraient encore dans l’usine."

generate_summary(text)

# Output: L’explosion a eu lieu dans l’usine Kristall à Dzerzhinsk, une ville située à environ 400 kilomètres à l’est de Moscou.

Manuel Romero/@mrm8488 创建,获得 Narrativa 的支持,用爱在西班牙制作