File size: 1,170 Bytes
2e0f9b2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 | ---
license: mit
library_name: transformers
tags:
- token-classification
- ner
- mountains
- bert
language:
- en
---
# Mountain NER (BERT fine-tuned)
Fine-tuned BERT token classifier for **mountain** named entity recognition
(`O`, `B-MOUNTAIN`, `I-MOUNTAIN`).
## Metrics (held-out test, 831 sentences)
| Metric | Value |
|--------|-------|
| Precision | 0.9448 |
| Recall | 0.9377 |
| F1 | **0.9413** |
Source: `Task_1_NLP/artifacts/eval/notebook_eval_results.csv` in the assessment repo.
## Load
```python
from transformers import AutoModelForTokenClassification, AutoTokenizer
repo = "<YOU>/mountain-ner-bert" # after upload
tok = AutoTokenizer.from_pretrained(repo)
model = AutoModelForTokenClassification.from_pretrained(repo)
```
Local path before upload: `publish/hf-task1-bert-ner/`.
## Training notes
- Started from a CoNLL-2003 NER BERT checkpoint and adapted to the mountain label set.
- Full pipeline: see `Task_1_NLP/` in the GitHub assessment repository.
- Intermediate Trainer checkpoints were **excluded** from this upload (final `model.safetensors` only).
## Citation / context
DS internship assessment — Task 1 (Mountain NER).
|