neural-backed-decision-trees icon indicating copy to clipboard operation
neural-backed-decision-trees copied to clipboard

Support for Continuous Data

Open eladmw opened this issue 3 years ago • 1 comments

This is pretty cool, I was just wondering if it also works on continuous data. I think resnet can work on continuous data, so I'm curious.

eladmw avatar Jan 13 '21 22:01 eladmw

@eladmw Thanks for asking -- do you mean regression? If so, not directly. However, if each of the leaves has a dedicated (possibly-learned) continuous value like in CART, you could reduce a regression problem to a classification problem. Then, use NBDT.

alvinwan avatar Jan 23 '21 06:01 alvinwan