From c6e8c89d39e58d57a16303b7d7dbae87d13ac3ae Mon Sep 17 00:00:00 2001
From: Jens Ahrensfeld <jens@jayfield.org>
Date: Tue, 11 Jan 2022 19:13:47 +0000
Subject: [PATCH] - improved test

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@787 b431acfa-c32f-4a4a-93f1-934dc6c82436
---
 source/main.cpp | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/source/main.cpp b/source/main.cpp
index 7319a77..48caa96 100644
--- a/source/main.cpp
+++ b/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;
 }