You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
The text was updated successfully, but these errors were encountered:
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.
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.
The text was updated successfully, but these errors were encountered: