This repo contains code for training a NER model for healthcare applications. The model is specifically trained on PHI records from i2b2 dataset . Source: https://www.i2b2.org/NLP/DataSets/

Models

The model.py file contains various model for constructing NER model including BERT model for HealthCare.

Train

The train.py file contains script for training the model

ner

The ner.py file contains code for preparing the data for the data for the NER model