Skip to content

Python library for custom entity recognition using Sklearn CRF

License

Notifications You must be signed in to change notification settings

ManikandanThangavelu/scikitcrf_NER

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Entity recognition using scikit CRF

Decscription

This is a simple python applicaion that uses sklearn-crfsuite for entity recognition using CRF.

Installation

Install this package using pip by running the follwing command

pip install scikitcrf_ner

if you face any issues while installing sklearn_crfsuite This may help

Make sure you download spacy english model using

python -m spacy download en

Usage

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 formatted Predict the entities by

entityRecognition.predict("Utterance")

Sample code

Refer this sample code:

from scikitcrf_ner import entityRecognition as ner
ner.train("sample_train.json")
entities = ner.predict("show me some Indian restaurants")
print(entites)

License

MIT