From 2ac8d27e75fa484182a3004b8ecd8e18f4e8b76d Mon Sep 17 00:00:00 2001
From: Jens Ahrensfeld <jens@jayfield.org>
Date: Mon, 28 Oct 2019 23:09:33 +0000
Subject: [PATCH] - better error report

git-svn-id: http://moon:8086/svn/software/trunk/projects/RBM@597 b431acfa-c32f-4a4a-93f1-934dc6c82436
---
 Source/Rbm.cpp | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/Source/Rbm.cpp b/Source/Rbm.cpp
index 7858895..ddc8039 100644
--- a/Source/Rbm.cpp
+++ b/Source/Rbm.cpp
@@ -261,11 +261,14 @@ void Rbm::train(size_t numEpochs, size_t miniBatchSize, bool &doStop)
 	
 	updateHiddenBatch();
 	
-	MatrixXd vis = m_h * m_w.weights().transpose();
+	MatrixXd h = __batch * m_w.weights();
+	h += m_w.hiddenBias().replicate(__batch.rows(), 1);
+	
+	MatrixXd vis = h * m_w.weights().transpose();
 	vis += m_w.visibleBias().replicate(__batch.rows(), 1);
 	MatrixXd diffErr = __batch - probsLogistic(vis);
 	diffErr.array() *= diffErr.array();
-	double err = diffErr.colwise().sum().mean();
+	double err = diffErr.array().mean();
 	cout << "error (total) = " << err << endl;
 
 	onProgressChanged();