inception-external-recommender icon indicating copy to clipboard operation
inception-external-recommender copied to clipboard

Add Transformer-based NER classifier using Hugging Face models

Open lfcc1 opened this issue 1 year ago • 4 comments

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 avatar Sep 20 '24 17:09 lfcc1

@lfcc1 build fails

reckart avatar Sep 20 '24 17:09 reckart

My bad. I commented some dependencies in the setup.py file for my specific use case. I have corrected it.

lfcc1 avatar Sep 20 '24 18:09 lfcc1

@lfcc1 Looks like there is still some problem with the requirements.

reckart avatar Sep 22 '24 11:09 reckart

@reckart Missed a comma, could you try again?

lfcc1 avatar Sep 23 '24 11:09 lfcc1

Thanks again :)

reckart avatar Oct 08 '24 07:10 reckart