search

alvinwan / neural-backed-decision-trees

Making decision trees competitive with neural networks on CIFAR10, CIFAR100, TinyImagenet200, Imagenet
https://nbdt.aaalv.in
MIT License
604 stars 130 forks source link

Support for Continuous Data #13

Open eladmw opened 3 years ago

eladmw commented 3 years ago

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.

alvinwan commented 3 years ago

@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.