It currently supports only English.
pip install -r requirements.txt
(optional)
jupyter notebook 01_train_models.ipynbMain program
python infer_model.pyThe script infer_model.py is an interactive CLI program.
After launching , you will be prompted to insert a sentence. The script outputs each inferred named entity for each token.
It includes a straightforward algorithm to reconstruct the original tokens by combining word-pieces, ensuring that each token is mapped to its corresponding entity.
ray_gridsearch.py was created for optimizing hyperparameters.
Always run your projects in a virtual environment.