Instructions to use ra1nbowdash/mountain-ner-bert with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use ra1nbowdash/mountain-ner-bert with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="ra1nbowdash/mountain-ner-bert")# Load model directly from transformers import AutoTokenizer, AutoModelForTokenClassification tokenizer = AutoTokenizer.from_pretrained("ra1nbowdash/mountain-ner-bert") model = AutoModelForTokenClassification.from_pretrained("ra1nbowdash/mountain-ner-bert", device_map="auto") - Notebooks
- Google Colab
- Kaggle
| 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). | |