GitHub - darth-c0d3r/deep-neural-decision-forests: A model combining Deep Neural Networks and (Stochastic) Random Forests.

darth-c0d3r / deep-neural-decision-forests Public

Notifications You must be signed in to change notification settings
Fork 5
Star 14

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

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
data/mnist		data/mnist
stats		stats
ndf.py		ndf.py
ndf.pyc		ndf.pyc
readme		readme
train.py		train.py

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