- improved test

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@787 b431acfa-c32f-4a4a-93f1-934dc6c82436

source/main.cpp

@@ -79,7 +79,6 @@ int main()
 
 	const string project("test");
 	
-	RbmListener statusDisplay;
 	Stack stack(".", project);
 
 	stack.loadTrainingBatch();
@@ -122,14 +121,17 @@ int main()
 	stack.loadWeights();
 #endif	
 	
+#if TRAIN_TEST
+	RbmListener statusDisplay;
+
 	// Train stack
 	stack.train(&statusDisplay);
 	
 	// Save weights
 	stack.saveWeights();
-
+#endif
 	Layer *layer = stack.getLayer(0);
-	arma::mat v = arma::randu(stack.trainingBatch().n_rows, layer->bv().n_elem);
+	arma::mat v = stack.trainingBatch();
 	arma::mat h = layer->toHiddenProbs(v);
 	arma::mat r = layer->toVisibleProbs(h);
 	
@@ -137,5 +139,9 @@ int main()
 	r = layer->upDownPass(v);
 	r.print("v2");
 	
+	arma::mat err = Rbm::rms_error(r-v);
+	err.print("err");
+	
+	
 	return 0;
 }