Skip to content

[Bonsai] Enable Regression for PyTorch#282

Open
siddhant-0707 wants to merge 2 commits intomicrosoft:masterfrom
siddhant-0707:dev/sidd-pytorch
Open

[Bonsai] Enable Regression for PyTorch#282
siddhant-0707 wants to merge 2 commits intomicrosoft:masterfrom
siddhant-0707:dev/sidd-pytorch

Conversation

@siddhant-0707
Copy link

@siddhant-0707 siddhant-0707 commented Nov 24, 2023

  • 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

@siddhant-0707
Copy link
Author

@microsoft-github-policy-service agree

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