Skip to content

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

Notifications You must be signed in to change notification settings

jackeown/tree2Net

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages