Repository files navigation

This is a simple python applicaion that usessklearn-crfsuite for entity recognition usingCRF.

Install this package using pip by running the follwing command

pip install scikitcrf_ner

Make sure you download spacy english model using

python -m spacy download en

Import the package using

from scikitcrf_ner import entityRecognition

Train the model using

entityRecognition.train("path/to/trainingfile.json")

Refer the sample training file(sample_train.json), the training file should be json formattedPredict the entities by

entityRecognition.predict("Utterance")

Refer this sample code:

fromscikitcrf_nerimportentityRecognitionasnerner.train("sample_train.json")entities=ner.predict("show me some Indian restaurants")print(entites)

MIT