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.

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

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

Merged
merged 10 commits into from
Sep 29, 2024

Conversation

lfcc1
Copy link
Contributor

@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
Contributor Author

lfcc1 commented Sep 20, 2024

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
Contributor Author

lfcc1 commented Sep 23, 2024

@reckart Missed a comma, could you try again?

@reckart reckart merged commit c683233 into inception-project:main Sep 29, 2024
4 checks passed
@reckart
Copy link
Member

reckart commented Oct 8, 2024

Thanks again :)

@lfcc1 lfcc1 deleted the transformers branch October 8, 2024 11:25
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
⭐️ Enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants