-
Notifications
You must be signed in to change notification settings - Fork 4.7k
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
DIETClassifier adds a confidence value to entity predictions #6038
Conversation
Tensorflow Addons PR: tensorflow/addons#1935 |
@tabergma could you please apply my comments to tf addons PR as well |
This is a good feature to add. However, current rasa documentation at https://rasa.com/docs/rasa/nlu/entity-extraction/#entity-extraction-custom-features says Does the documentation need to be updated to indicate that now DIET returns a confidence value? |
@vish0701 The feature is not yet released. It is merged into master and we also updated the documentation. Once we do the next release from master, you will have the confidence value for entities coming from the DIETClassifier and the documentation will be updated as well. |
Was this already included in a release? From my understanding it is not in 1.10.14. |
This will be included in the Rasa 2.0 release, which should happen any day now. |
Proposed changes:
DIETClassifier
now adds a confidence value to entity predictions.closes #5481
(Will also make a PR to tensorflow addons directly. If they take the changes and adapt the CRF, we can remove our custom CRF again.)
Status (please check what you already did):
black
(please check Readme for instructions)