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

Improve NN runtime by reducing num_categories_to_keep #246

Merged
merged 2 commits into from Jan 29, 2020
Merged

Conversation

jwmueller
Copy link
Contributor

@jwmueller jwmueller commented Jan 28, 2020

Tiny improvement to change default num_categories_to_keep hyperparameter of NN from 500 to 100, in order to improve runtimes.

Also allow neural net to be trained for 0 epochs (no training) which is useful for debugging and using just its data preprocessor.

@szha
Copy link

szha commented Jan 28, 2020

Job PR-246-1 is done.
Docs are uploaded to http://autogluon-staging.s3-website-us-west-2.amazonaws.com/PR-246/1/index.html

@szha
Copy link

szha commented Jan 29, 2020

Job PR-246-2 is done.
Docs are uploaded to http://autogluon-staging.s3-website-us-west-2.amazonaws.com/PR-246/2/index.html

@szha
Copy link

szha commented Jan 29, 2020

Job PR-246-3 is done.
Docs are uploaded to http://autogluon-staging.s3-website-us-west-2.amazonaws.com/PR-246/3/index.html

@jwmueller jwmueller merged commit 93a00ed into master Jan 29, 2020
@jwmueller jwmueller deleted the NNspeedup branch January 29, 2020 01:07
gradientsky pushed a commit to gradientsky/autogluon that referenced this pull request Jan 31, 2020
* changed default num_categories_to_keep hyperparameter of NN to improve runtimes

* allow NN to be trained for 0 epochs
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.

None yet

3 participants