Skip to content

unixpickle/neuraltree

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 

Repository files navigation

neuraltree

This is a differentiable decision tree with neural networks as branches and leaves. Trees of this form have more power than single layers of perceptrons, but I have yet to compare them to other deep architectures.

Algorithm

Each non-leaf node in a tree has a network and some children. The network produces a list of probabilities, one for each child. The leaf nodes have networks too, but these networks output final classifications. The goal when training the tree is to maximize the probability of outputting the correct classification. The tree can be trained easily using classical back propagation and a variant of Stochastic Gradient Descent.

About

Differentiable decision tree of neural networks

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages