Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add Transformer-based NER classifier using Hugging Face models #63

Open
wants to merge 10 commits into
base: main
Choose a base branch
from
Open

Add Transformer-based NER classifier using Hugging Face models #63

wants to merge 10 commits into from

Conversation

lfcc1
Copy link

@lfcc1 lfcc1 commented Sep 20, 2024

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

@reckart reckart added the ⭐️ Enhancement New feature or request label Sep 20, 2024
@reckart
Copy link
Member

reckart commented Sep 20, 2024

@lfcc1 build fails

@lfcc1
Copy link
Author

lfcc1 commented Sep 20, 2024 •
edited
Loading

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

@reckart
Copy link
Member

reckart commented Sep 22, 2024

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

@lfcc1
Copy link
Author

lfcc1 commented Sep 23, 2024

@reckart Missed a comma, could you try again?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees

@lfcc1 lfcc1

Labels
⭐️ Enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@lfcc1 @reckart