Instructions to use YarBar/bert-finetuned-ner-14-one-q with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use YarBar/bert-finetuned-ner-14-one-q with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="YarBar/bert-finetuned-ner-14-one-q")# Load model directly from transformers import AutoTokenizer, AutoModelForTokenClassification tokenizer = AutoTokenizer.from_pretrained("YarBar/bert-finetuned-ner-14-one-q") model = AutoModelForTokenClassification.from_pretrained("YarBar/bert-finetuned-ner-14-one-q") - Notebooks
- Google Colab
- Kaggle
bert-finetuned-ner-14-one-q
This model is a fine-tuned version of FacebookAI/roberta-base on an unknown dataset. It achieves the following results on the evaluation set:
- Loss: 0.0567
- Precision: 0.9276
- Recall: 0.9199
- F1: 0.9238
- Accuracy: 0.9838
Model description
More information needed
Intended uses & limitations
More information needed
Training and evaluation data
More information needed
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 2e-05
- train_batch_size: 16
- eval_batch_size: 128
- seed: 42
- optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
- lr_scheduler_type: linear
- num_epochs: 3
Training results
| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
|---|---|---|---|---|---|---|---|
| 0.1672 | 0.2278 | 200 | 0.1605 | 0.5846 | 0.4634 | 0.5170 | 0.9132 |
| 0.0841 | 0.4556 | 400 | 0.1075 | 0.7812 | 0.8314 | 0.8055 | 0.9615 |
| 0.0632 | 0.6834 | 600 | 0.0904 | 0.8100 | 0.8792 | 0.8432 | 0.9706 |
| 0.0486 | 0.9112 | 800 | 0.1480 | 0.8270 | 0.7760 | 0.8007 | 0.9553 |
| 0.0352 | 1.1390 | 1000 | 0.0756 | 0.8804 | 0.8709 | 0.8756 | 0.9735 |
| 0.0271 | 1.3667 | 1200 | 0.0775 | 0.8855 | 0.8859 | 0.8857 | 0.9762 |
| 0.0231 | 1.5945 | 1400 | 0.0604 | 0.9030 | 0.9245 | 0.9136 | 0.9839 |
| 0.0217 | 1.8223 | 1600 | 0.0674 | 0.9034 | 0.8895 | 0.8964 | 0.9768 |
| 0.0188 | 2.0501 | 1800 | 0.0649 | 0.9137 | 0.8995 | 0.9065 | 0.9793 |
| 0.0105 | 2.2779 | 2000 | 0.0529 | 0.9313 | 0.9306 | 0.9309 | 0.9865 |
| 0.0136 | 2.5057 | 2200 | 0.0513 | 0.9327 | 0.9298 | 0.9312 | 0.9858 |
| 0.0114 | 2.7335 | 2400 | 0.0456 | 0.9355 | 0.9416 | 0.9385 | 0.9881 |
| 0.0123 | 2.9613 | 2600 | 0.0562 | 0.9280 | 0.9207 | 0.9244 | 0.9839 |
| 0.0123 | 3.0 | 2634 | 0.0567 | 0.9276 | 0.9199 | 0.9238 | 0.9838 |
Framework versions
- Transformers 5.7.0
- Pytorch 2.10.0+cu128
- Datasets 4.8.5
- Tokenizers 0.22.2
- Downloads last month
- 43
Model tree for YarBar/bert-finetuned-ner-14-one-q
Base model
FacebookAI/roberta-base