Downloads · 30 days
16
3% of all-time downloads
large-BERT-NER-email
pascalrai/large-BERT-NER-email
large-BERT-NER-email is a token classification model from pascalrai. Use it when you need labels on individual words, such as names. It is set up for transformers.
All-time downloads
537
Public
Parameters
333M
2.7 GB on disk
Likes
1
Public
Repo makeup
Click a slice to open those files.
.safetensors1.3 GB · 100%
At a glance
- Task
- Token Classification
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Feb 15, 2024
- Updated
- Feb 15, 2024
- SHA
- 3bd867a2
Try a prompt
Model card
From the Hugging Face model README
Overview:
The Model is fine-tuned for 3 class + "0" class.<br> The Dataset is custom annotated and contains 400 texts and the model was trained on the split of 0.76, 0.12, and 0.12.
The validation classification report is as follows:
| Class | Precision | Recall | f1 |
|---|---|---|---|
| 0 | 1.00 | 1.00 | 1.00 |
| 1 | 0.98 | 1.00 | 0.91 |
| 2 | 0.95 | 0.89 | 0.92 |
| 3 | 0.8 | 0.88 | 0.84 |
| macro-avg | 0.93 | 0.94 | 0.94 |
The test classification report is as follows:
| Class | Precision | Recall | f1 |
|---|---|---|---|
| 0 | 1.00 | 1.00 | 1.00 |
| 1 | 0.98 | 1.00 | 0.99 |
| 2 | 0.66 | 0.97 | 0.79 |
| 3 | 0.84 | 0.78 | 0.81 |
| macro-avg | 0.87 | 0.94 | 0.90 |
Possible future direction:
- Clean data to a good enough format as much as possible.
- Increase the data as much as possible. (Make sure to have data that is seen in real use cases.)
- Ponder: Is it possible to use sth like Grammarly to clean the sentences before tokenization such that proper nouns are Capital and the grammer is correct such that a pattern is formed?
- Task
- Token Classification
- Library
- transformers
- Type
- bert
- Created
- Feb 15, 2024
- Updated
- Feb 15, 2024