- refactored

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@751 b431acfa-c32f-4a4a-93f1-934dc6c82436
2022-01-08 10:41:41 +00:00
parent 9472bb0a36
commit 637b237966
4 changed files with 39 additions and 28 deletions
@@ -67,6 +67,16 @@ public:
 		return m_bh.n_elem;
 	}

+      	arma::mat toHiddenProbs(const arma::mat &visible) const
+        {
+            return Rbm::prob(v_to_h(visible));
+        }
+
+      	arma::mat toVisibleProbs(const arma::mat &hidden) const
+        {
+            return Rbm::prob(h_to_v(hidden));
+        }
+
 	arma::mat trainingData(arma::mat const &batch)
 	{
 		arma::mat thisBatch = batch;