Downloads · 30 days
8
3% of all-time downloads
bert-english-ner-mountain
Strongich/bert-english-ner-mountain
bert-english-ner-mountain is a token classification model from Strongich. Use it when you need labels on individual words, such as names. It is set up for transformers. The card lists the license as apache-2.0.
This model is fine-tuned https://huggingface.co/dslim/bert-base-NER for mountain names recognition. Keep in mind, that this model uses 2 less layers, than the original due to lack of GPU memory. \ You can find more in…
All-time downloads
312
Public
Parameters
93.5M
374 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors374 MB · 100%
At a glance
- Task
- Token Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- bert
- Access
- Public
- Created
- Dec 7, 2023
- Updated
- Dec 7, 2023
- SHA
- 5db05493
Try a prompt
Model card
From the Hugging Face model README
About
This model is fine-tuned https://huggingface.co/dslim/bert-base-NER for mountain names recognition. Keep in mind, that this model uses 2 less layers, than the original due to lack of GPU memory.
You can find more info in my github repo
Dataset
It was trained using ~150 samples of created dataset by ChatGPT in NER type.
How to use
You can use this model with Transformers pipeline for NER.
from transformers import AutoTokenizer, AutoModelForTokenClassification
from transformers import pipeline
tokenizer = AutoTokenizer.from_pretrained('Strongich/bert-english-ner-mountain')
model = AutoModelForTokenClassification.from_pretrained('Strongich/bert-english-ner-mountain')
nlp = pipeline("ner", model=model, tokenizer=tokenizer)
example = "The sun dipped below the horizon, casting a warm glow upon the rugged slopes of Mount Everest."
ner_results = nlp(example)
print(ner_results)
- Task
- Token Classification
- Library
- transformers
- Type
- bert
- License
- apache-2.0
- Created
- Dec 7, 2023
- Updated
- Dec 7, 2023