- moved numEpochs and miniBatchSize to RBM::Params

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@606 b431acfa-c32f-4a4a-93f1-934dc6c82436
2019-11-06 06:12:51 +00:00
parent c834a86f0f
commit 3ef7b07f5e
6 changed files with 31 additions and 15 deletions
@@ -101,6 +101,9 @@ int main()
 		stack.addLayer(layer);
 	}

+	Layer *layer = stack.getLayer(0);
+	layer->params().learningRate = 0.02;
+
 	// Save project
 	stack.save();

@@ -119,12 +122,11 @@ int main()
 #endif	
 	
 	// Train stack
-	stack.train(batch, 100, 1000, &statusDisplay);
-
+	stack.train(batch, &statusDisplay);
+	
 	// Save weights
 	stack.saveWeights();

-	Layer *layer = stack.getLayer(0);
 	arma::mat v = arma::randu(numTraining, layer->bv().n_elem);
 	arma::mat h = layer->toHiddenProbs(v);
 	arma::mat r = layer->toVisibleProbs(h);