erika1203 / mydbn Goto Github PK

This deep belief network code need numpy module.
Please unzip data.zip which involves the weights of 2 rbm layers and softmax as well as the training and test dataset.
How to execute the programme?
train => python dbn.py train
test  => python dbn.py test

Where's the test and train come from?
They're part of the data from MNIST handwritten digit database.You can use preparedata to extract part of the data from it.