Views
No views yet
DistilBertForTokenClassification for adding punctuations to plain text (uncased English)1from transformers import DistilBertForTokenClassification, DistilBertTokenizerFast
2
3model = DistilBertForTokenClassification.from_pretrained("Qishuai/distilbert_punctuator_en")
4tokenizer = DistilBertTokenizerFast.from_pretrained("Qishuai/distilbert_punctuator_en")| precision | recall | f1-score | support | |
|---|---|---|---|---|
| COMMA | 0.66 | 0.55 | 0.60 | 7064 |
| EXLAMATIONMARK | 1.00 | 0.00 | 0.00 | 5 |
| PERIOD | 0.73 | 0.63 | 0.68 | 6573 |
| QUESTIONMARK | 0.54 | 0.41 | 0.47 | 17 |
| micro avg | 0.69 | 0.59 | 0.64 | 13659 |
| macro avg | 0.73 | 0.40 | 0.44 | 13659 |
| weighted avg | 0.69 | 0.59 | 0.64 | 13659 |
| precision | recall | f1-score | support | |
|---|---|---|---|---|
| COMMA | 0.71 | 0.56 | 0.63 | 10712 |
| EXLAMATIONMARK | 0.45 | 0.07 | 0.12 | 75 |
| PERIOD | 0.75 | 0.65 | 0.70 | 7921 |
| QUESTIONMARK | 0.73 | 0.67 | 0.70 | 827 |
| micro avg | 0.73 | 0.60 | 0.66 | 19535 |
| macro avg | 0.66 | 0.49 | 0.53 | 19535 |
| weighted avg | 0.73 | 0.60 | 0.66 | 19535 |