- changed parameter order of train()

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@571 b431acfa-c32f-4a4a-93f1-934dc6c82436
2019-10-24 18:54:40 +00:00
parent d6ed27f5d2
commit 0e544c1afc
3 changed files with 3 additions and 3 deletions
@@ -120,7 +120,7 @@ int main()
 	arma::mat bv = arma::zeros(1, numVisible);
 	arma::mat bh = arma::zeros(1, numHidden);
 	Rbm rbm(params, w, bv, bh);
-	rbm.train(batch, 100, 100, &statusDisplay);
+	rbm.train(batch, 1000, 100, &statusDisplay);
 	saveWeight("mnist_2.weights.dat", 28, 28, w, bv, bh);
 	arma::mat v = arma::randu(numTraining, numVisible);
 	arma::mat h = rbm.toHiddenProbs(v);