dieumerci/mountain-recognition-ner

Mountain Name Recognition (NER) Model

Overview

This repository contains a pre-trained Named Entity Recognition (NER) model for mountain name recognition. The model is based on the dslim/bert-large-NER architecture and was fine-tuned on a relabeled subset of the DFKI-SLT/few-nerd dataset. The model achieves an F1 Score of 87.42% on the test set.

Inference

Here's a sample code for performing inference using the model:

from transformers import AutoModelForTokenClassification, AutoTokenizer, pipeline

model = AutoModelForTokenClassification.from_pretrained("dieumerci/mountain-recognition-ner")
tokenizer = AutoTokenizer.from_pretrained("dieumerci/mountain-recognition-ner")
classifier = pipeline("ner", model=model, tokenizer=tokenizer)

text = "Next on our list is Denali Peak, also known as Mount McKinley, in Alaska."
result = classifier(text)

print(result)