-
Notifications
You must be signed in to change notification settings - Fork 24
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
NLPCRAFT-11: Add auto-enrich using Bert and FastTest models #1
Conversation
@Ifropc Thank you very much for sharing, this feature looks very promising! I would suggest the following improvements:
|
Add apache license Replace prints with logging Move printing graphs to jupyter
Hi @Ifropc ! Thank your for your work, I have tried it and results are very impressive! Below some remarks for discussion:
Let's discuss! |
Just for main cases (empty request, invalid indexes range) - now result is code 500, internal error. |
- Added validation, limit argument
Hi @skhdl , thanks you for your feedback.
Also I upgraded torch to latest version (1.5), please re-run install script. |
I'm closing this pull request. Further work would be done in branch NLPCRAFT-67 |
This pull request should resolve NLPCRAFT-11: auto-enrich user models with synonyms
Proposed approach uses Bert (RoBerta) model to generate synonyms for given context, masking target word. Afterwards, output is filtered with FastTest for specified context.
This feature could also be integrated with NLPCRAFT-41