[Bonsai] Enable Regression for PyTorch #282

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.

Sign up for GitHub

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

Open

siddhant-0707 wants to merge 2 commits into microsoft:master from siddhant-0707:dev/sidd-pytorch

siddhant-0707 commented Nov 24, 2023 •

edited

Loading

Added regression capability for pytorch implementation of Bonsai
Similar to tf implementation, you can train model specifically for regression using -regression True flag
Tested on boston-housing and mg regression datasets

Hey @harsha-simhadri, @aayan636 please take a look


          enabled regression

40cee4a

Author

siddhant-0707 commented Nov 24, 2023

@microsoft-github-policy-service agree


          tests

bfcfd17

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

Labels

None yet