A simple NER training task with an invoice extraction dataset.

Data, library preparation

Download the Multi-layout Invoice Document Dataset (MIDD) for exmaple from this link, and put IOB to data directory

pip install -r requirements.txt

Train the model

python train.py

NER on test data

python ner_test.py

You will get results like this.