Skip to content

Latest commit

 

History

History
10 lines (7 loc) · 359 Bytes

README.md

File metadata and controls

10 lines (7 loc) · 359 Bytes

tree2Net

Convert a sci-kit learn decision tree to a simple neural network in a straightforward way!

tree2Pytorch is a good example of how to use tree2nn with pytorch. It should be pretty easy to use with Keras as well.

old_stuff contains original tensorflow code which is now obsolete.

Usage is explained in comments in the tree2Net.py