inception-external-recommender
inception-external-recommender copied to clipboard
Add Transformer-based NER classifier using Hugging Face models
What's in the PR
- Added a TransformerNerClassifier class that allows to perform Named Entity Recognition using pre-trained models from Hugging Face's transformers.
How to test
- Run test file: pytest tests/test_transformer_recommender.py
@lfcc1 build fails
My bad. I commented some dependencies in the setup.py file for my specific use case. I have corrected it.
@lfcc1 Looks like there is still some problem with the requirements.
@reckart Missed a comma, could you try again?
Thanks again :)