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

1 participant