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

Published 20 hours ago • verified publisher icon alvinwan

Tensorflow support

Open oak-tree opened this issue 4 years ago • 3 comments

Hey, This looks like a great project, Does it support or will support Tensorflow engine as well? ?

oak-tree avatar May 12 '20 06:05 oak-tree

Thanks -- unfortunately, it does not support Tensorflow, but PRs are welcome!

alvinwan avatar May 12 '20 16:05 alvinwan

@alvinwan Thanks for the answer, if you can guide me where are the dependencies on PyTorch it can save lots of time :)

oak-tree avatar May 15 '20 11:05 oak-tree

@oak-tree Hope this helps: Model-related wrappers are in nbdt/model.py. The data structure in data/custom.py and utils are framework-agnostic though. Beyond that, unfortunately, I'm not familiar with Tensorflow so don't know what's PyTorch specific.

alvinwan avatar May 15 '20 20:05 alvinwan