EdgeML icon indicating copy to clipboard operation
EdgeML copied to clipboard
verified publisher icon microsoft

[Bonsai] Enable Regression for PyTorch

Open siddhant-0707 opened this issue 2 years ago • 1 comments

  • 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 avatar Nov 24 '23 13:11 siddhant-0707

@microsoft-github-policy-service agree

siddhant-0707 avatar Nov 24 '23 13:11 siddhant-0707