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

Jump to bottom

[FR] Argmax each batch in predict_classes #5081

Closed
ohadle opened this issue Jan 19, 2017 · 0 comments
Closed

[FR] Argmax each batch in predict_classes #5081

ohadle opened this issue Jan 19, 2017 · 0 comments

Comments

@ohadle
Copy link

ohadle commented Jan 19, 2017

Currently, predict_classes first calculates probabilities for all the data and then performs the argmax. For large datasets, with many softmax bins, this is not memory-efficient.
Instead, keras could perform the argmax per-batch.
@stale stale bot added the stale label May 23, 2017
@stale stale bot closed this as completed Jun 22, 2017
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@ohadle