Skip to content

add method to parse with probabilities#1

Open
nkansal96 wants to merge 1 commit intomacournoyer:masterfrom
nkansal96:feature/parse-with-probabilities
Open

add method to parse with probabilities#1
nkansal96 wants to merge 1 commit intomacournoyer:masterfrom
nkansal96:feature/parse-with-probabilities

Conversation

@nkansal96
Copy link

Added a method parse_with_probabilities. This returns a hash of all classes with respective probabilities (instead of just the max_class). This would be useful when trying to determine the certainty of a prediction (or getting next-best results).

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant