GitHub - nasame/MyDBN: The implementation of DBN ( RBM) and softmax regression

nasame / MyDBN Public

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

The implementation of DBN ( RBM) and softmax regression

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 1 Commit
common.py		common.py
data.zip		data.zip
dbn.py		dbn.py
preparedata.py		preparedata.py
rbm.py		rbm.py
readme.txt		readme.txt
softmax.py		softmax.py

Repository files navigation

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.