Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.

Support more return types in predict #191

Closed
edenlightning opened this issue Mar 25, 2021 · 0 comments · Fixed by #229
Closed

Support more return types in predict #191

edenlightning opened this issue Mar 25, 2021 · 0 comments · Fixed by #229
Labels
enhancement New feature or request help wanted Extra attention is needed

Comments

@edenlightning
Copy link
Contributor

Currently, the predict function returns only categories.

As a user I want to set the return type in the predict function, so I can get a label or raw probabilities instead of just the category.
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Assignees
No one assigned
Labels
enhancement New feature or request help wanted Extra attention is needed
Projects
None yet
Milestone
No milestone
1 participant
@edenlightning