jens/Rbm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

- improved RBM

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@567 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2019-10-24 18:22:30 +00:00
parent 0c71b2e6b8
commit 5b4aed047f
3 changed files with 60 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/main.cpp
+3 -2
View File
@@ -16,6 +16,7 @@ class RbmListener : public Rbm::IListener
std::cout << "epoch : " << status.epoch << std::endl;
std::cout << "trainingSizeRemain: " << status.trainingSizeRemain << std::endl;
std::cout << "error (per mini batch) = " << status.err << std::endl;
std::cout << "error (total) = " << status.err_total << std::endl;
std::cout << "L1 = " << status.L1 << std::endl;
std::cout << "L2 = " << status.L2 << std::endl;
@@ -122,7 +123,7 @@ int main()
rbm.train(batch, 100, 100, &statusDisplay);
saveWeight("mnist_2.weights.dat", 28, 28, w, bv, bh);
arma::mat v = arma::randu(numTraining, numVisible);
arma::mat h = rbm.toHidden(v);
arma::mat r = rbm.toVisible(h);
arma::mat h = rbm.toHiddenProbs(v);
arma::mat r = rbm.toVisibleProbs(h);
return 0;
}