Custom Named-Entity-Recognition

About

This is an implementation of the Custom Named-Entity-Recognition which applies BERT.
The objective is to find useful information present in resume like Education, Skills, Experience, etc.
BERT - https://github.com/google-research/bert#what-is-bert

Links

Dataset - (https://www.kaggle.com/dataturks/resume-entities-for-ner)

Install venv and activate

pip install virtualenv
virtualenv venv
cd venv/Scripts
activate 
OR 
source venv/bin/activate

Install requirements

pip install -r req.txt

Training

To train the model, Run:

python train.py

optional arguments:

-e epochs         Number of Epochs

-o path             Output path to save the trained model

Flask REST API

python app.py

API will be live at endpoints : https://localhost:5000/predict/

Test the API

Either use cURL, Postman or any other REST client

curl --location --request POST "http://localhost:5000/predict" --form "resume=@/resume-path.pdf"

NER Demo result

For bugs and suggestions, please raise a pull request or drop me a mail at sumedh.khodke@gmail.com