My favorites | Sign in
Project Home Downloads Wiki Issues Source
Project Information
Members
Featured
Downloads
Links

This is a small library that can train Restricted Boltzmann Machines, and also Deep Belief Networks of stacked RBM's.

Train RBM's:

%train an RBM with binary visible units and 500 binary hidden
model= rbmBB(data, 500);

%visualize the learned weights
visualize(model.W);

Do classification:

model= rbmFit(data, 500, labels);
prediction= rbmPredict(model, testdata);

Train a Deep Belief Network with 500,500,2000 architecture for classification:

models= dbnFit(data, [500 500 2000], labels);
prediction= dbnPredict(models, testdata);

see included example code for more

I can be contacted on andrej.karpathy@ gmail.

Powered by Google Project Hosting