address-ner-ru

Address NER model to find address parts from string

https://huggingface.co/aidarmusin/address-ner-ru

Dataset

5K raw addresses dataset

90% for training and 10% for evaluation

Evaluation

Example

from transformers import pipeline
import torch
import logging

device = "cuda:0" if torch.cuda.is_available() else "cpu"
logging.info(f"using device: {device}")

address_ner_pipeline = pipeline("ner", model="aidarmusin/address-ner-ru", device=device)

address = "628672,,,, Автономный Округ Ханты-Мансийский Автономный Округ - Югра,, Г. Лангепас, Ул. Солнечная, Д.21"
entities = address_ner_pipeline(address)
print(entities)