Tensorflow support

[oak-tree]

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

[alvinwan]

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

[oak-tree]

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

[alvinwan]

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