EdgeML icon indicating copy to clipboard operation
EdgeML copied to clipboard

Added Heirarchical prior.

Open meetdoshi90 opened this issue 4 years ago • 3 comments

Added path smoothing to prevent overfitting for deeper trees whilst improving accuracy. Added mse loss for regression on weighted layers. References:

  1. Path smoothing
  2. IHT Gradient descent

meetdoshi90 avatar Aug 31 '21 10:08 meetdoshi90

CLA assistant check
All CLA requirements met.

ghost avatar Aug 31 '21 10:08 ghost

hi @meetdoshi90,

This is very useful and seems correct. Thanks for the contribution. However, I think the TF bonsai version is obsolete due to its old dependencies. I think more people use the pytorch version more often.

@harsha-simhadri you might want to take a look at this when you get a chance.

Aditya

adityakusupati avatar Sep 02 '21 18:09 adityakusupati

Hey @adityakusupati I will implement the same on pytorch. Thanks for the advice. Meet.

meetdoshi90 avatar Sep 03 '21 04:09 meetdoshi90