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

Get probability distribution over all classes in Inference mode #102

Closed
Alex-Zien opened this issue Sep 25, 2019 · 1 comment
Closed

Get probability distribution over all classes in Inference mode #102

Alex-Zien opened this issue Sep 25, 2019 · 1 comment
Assignees
Labels
enhancement New feature or request

Comments

@Alex-Zien
Copy link

When we use the inferencer to get predictions it might be useful to return probability values for all labels, not just the maximum. My use case would be for NER, but I see it being useful for other classification tasks as well.

@Timoeller
Copy link
Contributor

Hey @Alex-Zien Thanks for creating this issue.
We realized it is a very important feature and super easy to implement, so we pushed it into master. Have a look at #103
Please update your FARM. You can now return class probs with setting the return_class_probs parameter to True in the Inferencer.
Hope that helps.

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

No branches or pull requests

2 participants