From 9a381c1899157e4abf5c494a1b9b02012b52a683 Mon Sep 17 00:00:00 2001 From: jens Date: Wed, 24 Jan 2024 15:14:30 +0100 Subject: [PATCH] - its gaussion hidden (not gaussian visible) --- source/Rbm.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/source/Rbm.cpp b/source/Rbm.cpp index 5da34b1..ae05cf4 100644 --- a/source/Rbm.cpp +++ b/source/Rbm.cpp @@ -131,13 +131,13 @@ void Rbm::gibbs_hv(arma::mat &h_probs, arma::mat &v_probs) const void Rbm::contrastiveDivergence(arma::mat const &v_states, arma::mat &dw, arma::mat &dbhv, arma::mat &dbv) { arma::mat v_probs(v_states); - arma::mat h_states; + arma::mat h_states = v_to_h(v_states); arma::mat h_probs; // Sample hidden - if (m_params.doGaussianVisible) + if (m_params.doGaussianHidden) { - h_probs = v_to_h(v_states); + h_probs = h_states; h_states = h_probs + arma::randn(h_probs.n_rows, h_probs.n_cols); } else if (m_params.doRaoBlackwell)