中文

基于 BERT 模型的自动标点恢复

计算与语言 2021-01-20 v1

摘要

我们提出了一种利用 BERT 模型进行英语与匈牙利语自动标点恢复的方法。对于英语,我们在 Ted Talks(标点恢复的常用基准)上开展实验;对于匈牙利语,我们在 Szeged Treebank 数据集上评估模型。我们的最佳模型在英语和匈牙利语上分别取得了 79.8 和 82.2 的宏平均 F1F_1 分数。我们的代码已公开可用。

关键词

引用

@article{arxiv.2101.07343,
  title  = {Automatic punctuation restoration with BERT models},
  author = {Attila Nagy and Bence Bial and Judit Ács},
  journal= {arXiv preprint arXiv:2101.07343},
  year   = {2021}
}

备注

11 pages, 6 figures, source code at https://github.com/attilanagy234/neural-punctuator