Mbart Large Cc25 Cnn Dailymail Xsum Nl

🚀 mbart-large-cc25-cnn-dailymail-xsum-nl

本项目是一个用于荷兰语新闻文章摘要生成的模型，基于微调后的 mbart 模型，能有效帮助用户快速获取文章核心内容。

🚀 快速开始

模型描述

这是 mbart 的微调版本。我们还在 此处 撰写了一篇关于该模型的 博客文章。

预期用途和限制

该模型旨在对荷兰语新闻文章进行摘要生成。

如何使用

import transformers

undisputed_best_model = transformers.MBartForConditionalGeneration.from_pretrained(
    "ml6team/mbart-large-cc25-cnn-dailymail-xsum-nl"
)
tokenizer = transformers.MBartTokenizer.from_pretrained("facebook/mbart-large-cc25")
summarization_pipeline = transformers.pipeline(
    task="summarization",
    model=undisputed_best_model,
    tokenizer=tokenizer,
)
summarization_pipeline.model.config.decoder_start_token_id = tokenizer.lang_code_to_id[
    "nl_XX"
]

article = "Kan je dit even samenvatten alsjeblief."  # Dutch
summarization_pipeline(
    article,
    do_sample=True,
    top_p=0.75,
    top_k=50,
    min_length=50,
    early_stopping=True,
    truncation=True,
)[0]["summary_text"]

📦 安装指南

文档未提及具体安装步骤，可参考 transformers 库的官方安装说明进行安装。

🔧 技术细节

训练数据

使用 此数据集 和 此数据集 对 mbart 进行微调。