模型:
oliverguhr/fullstop-dutch-punctuation-prediction
这个模型预测荷兰文本的标点符号。我们开发它来恢复转录的口语语言的标点符号。该模型是在 Europarl Dataset 上训练的。模型恢复以下标点标记:"。",",","?","-",":"
我们提供了一个简单的Python包,允许您处理任意长度的文本。
要开始安装包,请从 pypi 安装:
pip install deepmultilingualpunctuation
from deepmultilingualpunctuation import PunctuationModel model = PunctuationModel(model="oliverguhr/fullstop-dutch-punctuation-prediction") text = "hervatting van de zitting ik verklaar de zitting van het europees parlement die op vrijdag 17 december werd onderbroken te zijn hervat" result = model.restore_punctuation(text) print(result)
输出
hervatting van de zitting ik verklaar de zitting van het europees parlement, die op vrijdag 17 december werd onderbroken, te zijn hervat。
from deepmultilingualpunctuation import PunctuationModel model = PunctuationModel(model="oliverguhr/fullstop-dutch-punctuation-prediction") text = "hervatting van de zitting ik verklaar de zitting van het europees parlement die op vrijdag 17 december werd onderbroken te zijn hervat" clean_text = model.preprocess(text) labled_words = model.predict(clean_text) print(labled_words)
输出
[['hervatting', '0', 0.9999777], ['van', '0', 0.99998415], ['de', '0', 0.999987], ['zitting', '0', 0.9992779], ['ik', '0', 0.9999889], ['verklaar', '0', 0.99998295], ['de', '0', 0.99998856], ['zitting', '0', 0.9999895], ['van', '0', 0.9999902], ['het', '0', 0.999992], ['europees', '0', 0.9999924], ['parlement', ',', 0.9915131], ['die', '0', 0.99997807], ['op', '0', 0.9999882], ['vrijdag', '0', 0.9999746], ['17', '0', 0.99998784], ['december', '0', 0.99997866], ['werd', '0', 0.9999888], ['onderbroken', ',', 0.99287957], ['te', '0', 0.9999864], ['zijn', '0', 0.99998176], ['hervat', '.', 0.99762934]]
对于单个标点符号,如连字符和冒号,性能会有所不同,很多情况下,这些标点符号是可选的,并且可以用逗号或句号替换。该模型实现以下F1得分:
Label | Dutch |
---|---|
0 | 0.993588 |
. | 0.961450 |
? | 0.848506 |
, | 0.810883 |
: | 0.655212 |
- | 0.461591 |
macro average | 0.788538 |
micro average | 0.983492 |
@misc{https://doi.org/10.48550/arxiv.2301.03319, doi = {10.48550/ARXIV.2301.03319}, url = {https://arxiv.org/abs/2301.03319}, author = {Vandeghinste, Vincent and Guhr, Oliver}, keywords = {Computation and Language (cs.CL), Artificial Intelligence (cs.AI), FOS: Computer and information sciences, FOS: Computer and information sciences, I.2.7}, title = {FullStop:Punctuation and Segmentation Prediction for Dutch with Transformers}, publisher = {arXiv}, year = {2023}, copyright = {Creative Commons Attribution Share Alike 4.0 International} }