|
Project Information
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. |