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