/train-custom-ner

Train a custom NER model with spaCy - Workshop

Primary LanguagePython

Train a custom NER model with spaCy

Open in Streamlit visitor badge

In this workshop you will learn to annotate and prepare text data to train a custom NER model with spaCy. The provided dataset is too small to have good results, however the same approach can be implemented on a larger dataset. The pre-defined entity types (classes or categories) are: PERSON and LOC.

flowchart LR

A[Prepare and Annotate data] --> B[Prepare NER pipeline]
B --> C[Train NER model]
C --> D[Evaluate model]
D --> E[Test the model]
Loading