From 675a5485fe5c1b762381998cc4cd6feaee32d627 Mon Sep 17 00:00:00 2001
From: Jens Ahrensfeld <jens@jayfield.org>
Date: Tue, 22 Oct 2019 06:20:43 +0000
Subject: [PATCH] . improved toHidden(), toVisible()

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

diff --git a/source/Rbm.cpp b/source/Rbm.cpp
index b99242a..cf1e698 100644
--- a/source/Rbm.cpp
+++ b/source/Rbm.cpp
@@ -184,14 +184,14 @@ arma::mat Rbm::sample(const arma::mat &src)
 
 arma::mat Rbm::toHidden(const arma::mat &visible)
 {
-	arma::mat h = visible.t() * m_w  + m_bh;
-	return h;
+	arma::mat h = visible * m_w  + arma::repmat(m_bh, visible.n_rows, 1);
+	return probsLogistic(h);
 }
 
 arma::mat Rbm::toVisible(const arma::mat &hidden)
 {
-	arma::mat v = hidden * m_w.t() + m_bv;
-	return v;
+	arma::mat v = hidden * m_w.t() + arma::repmat(m_bv, hidden.n_rows, 1);
+	return probsLogistic(v);
 }
 
 void Rbm::uniform(arma::mat& srcDst, double mu, double stdDev)
diff --git a/source/Rbm.hpp b/source/Rbm.hpp
index 530d5b4..cced10b 100644
--- a/source/Rbm.hpp
+++ b/source/Rbm.hpp
@@ -65,7 +65,7 @@ public:
 		}
 	};
 
-	Rbm(size_t numHidden, size_t numVisible);
+	Rbm(size_t numVisible, size_t numHidden);
 	Rbm(const Rbm& orig);
 	virtual ~Rbm();
 
diff --git a/source/main.cpp b/source/main.cpp
index 128c865..7502023 100644
--- a/source/main.cpp
+++ b/source/main.cpp
@@ -56,10 +56,10 @@ int main()
 	Rbm::Params params;
 	Rbm rbm(28*28, 64);
 	arma::mat batch = arma::randu(100, 28*28);
-	rbm.train(batch, 100, 10, params, &statusDisplay);
+	rbm.train(batch, 10, 10, params, &statusDisplay);
 	
-//	arma::mat v = arma::randu(28*28, 1);
-//	arma::mat h = rbm.toHidden(v);
-//	arma::mat r = rbm.toVisible(h);
+	arma::mat v = arma::randu(10, 28*28);
+	arma::mat h = rbm.toHidden(v);
+	arma::mat r = rbm.toVisible(h);
 	return 0;
 }
\ No newline at end of file