Skip to content

A model combining Deep Neural Networks and (Stochastic) Random Forests.

Notifications You must be signed in to change notification settings

darth-c0d3r/deep-neural-decision-forests

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

This is an implementation of Deep Neural Decision Forests in PyTorch.
I have only added the database loader for MNIST Dataset. 
Thanks to https://github.com/jingxil/Neural-Decision-Forests,
from where the base code was taken.
Also, the changes required due the change in PyTorch API have been made.
I have added Batch Normalisation Layers to analyse the change in performance.

Read my analysis of the papers and the summary of my work which
will aid in further understanding of the work.
Link: https://medium.com/@gurparkashsinghsohi/deep-neural-decision-forests-b1dd39c4c6ce

About

A model combining Deep Neural Networks and (Stochastic) Random Forests.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages