Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
b3f272ff
Commit
b3f272ff
authored
Apr 27, 2020
by
Manuel Romero
Committed by
Julien Chaumond
Apr 27, 2020
Browse files
Create model card
parent
518f291e
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
72 additions
and
0 deletions
+72
-0
model_cards/mrm8488/bert-small-finetuned-typo-detection/README.md
...rds/mrm8488/bert-small-finetuned-typo-detection/README.md
+72
-0
No files found.
model_cards/mrm8488/bert-small-finetuned-typo-detection/README.md
0 → 100644
View file @
b3f272ff
---
language
:
english
thumbnail
:
---
# BERT SMALL + Typo Detection ✍❌✍✔
[
BERT SMALL
](
https://huggingface.co/google/bert_uncased_L-4_H-512_A-8
)
fine-tuned on
[
GitHub Typo Corpus
](
https://github.com/mhagiwara/github-typo-corpus
)
for
**typo detection**
(using
*NER*
style)
## Details of the downstream task (Typo detection as NER)
-
Dataset:
[
GitHub Typo Corpus
](
https://github.com/mhagiwara/github-typo-corpus
)
📚
-
[
Fine-tune script on NER dataset provided by Huggingface
](
https://github.com/huggingface/transformers/blob/master/examples/run_ner.py
)
🏋️♂️
## Metrics on test set 📋
| Metric | # score |
| :-------: | :-------: |
| F1 |
**89.12**
|
| Precision |
**93.82**
|
| Recall |
**84.87**
|
## Model in action 🔨
Fast usage with
**pipelines**
🧪
```
python
from
transformers
import
pipeline
typo_checker
=
pipeline
(
"ner"
,
model
=
"mrm8488/bert-small-finetuned-typo-detection"
,
tokenizer
=
"mrm8488/bert-small-finetuned-typo-detection"
)
result
=
typo_checker
(
"here there is an error in coment"
)
result
[
1
:
-
1
]
# Output:
[{
'entity'
:
'ok'
,
'score'
:
0.9021041989326477
,
'word'
:
'here'
},
{
'entity'
:
'ok'
,
'score'
:
0.7975626587867737
,
'word'
:
'there'
},
{
'entity'
:
'ok'
,
'score'
:
0.8596242070198059
,
'word'
:
'is'
},
{
'entity'
:
'ok'
,
'score'
:
0.7071516513824463
,
'word'
:
'an'
},
{
'entity'
:
'ok'
,
'score'
:
0.943381130695343
,
'word'
:
'error'
},
{
'entity'
:
'ok'
,
'score'
:
0.8047608733177185
,
'word'
:
'in'
},
{
'entity'
:
'ok'
,
'score'
:
0.8240702152252197
,
'word'
:
'come'
},
{
'entity'
:
'typo'
,
'score'
:
0.5004884004592896
,
'word'
:
'##nt'
}]
```
It works🎉! we typed
```coment```
instead of
```comment```
Let's try with another example
```
python
result
=
typo_checker
(
"Adddd validation midelware"
)
result
[
1
:
-
1
]
# Output:
[{
'entity'
:
'ok'
,
'score'
:
0.7128152847290039
,
'word'
:
'add'
},
{
'entity'
:
'typo'
,
'score'
:
0.5388424396514893
,
'word'
:
'##dd'
},
{
'entity'
:
'ok'
,
'score'
:
0.94792640209198
,
'word'
:
'validation'
},
{
'entity'
:
'typo'
,
'score'
:
0.5839331746101379
,
'word'
:
'mid'
},
{
'entity'
:
'ok'
,
'score'
:
0.5195121765136719
,
'word'
:
'##el'
},
{
'entity'
:
'ok'
,
'score'
:
0.7222476601600647
,
'word'
:
'##ware'
}]
```
Yeah! We typed wrong
```Add and middleware```
> Created by [Manuel Romero/@mrm8488](https://twitter.com/mrm8488)
> Made with <span style="color: #e25555;">♥</span> in Spain
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment