From be9a4a4d74f9d3afaa6f7b8ca1cb44311dc08fda Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Sat, 8 Jan 2022 10:56:36 +0000 Subject: [PATCH] - refactored git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@752 b431acfa-c32f-4a4a-93f1-934dc6c82436 --- source/Rbm.cpp | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/source/Rbm.cpp b/source/Rbm.cpp index fbe7c55..1c11ca3 100644 --- a/source/Rbm.cpp +++ b/source/Rbm.cpp @@ -73,13 +73,17 @@ Json::Value Rbm::toJson() const void Rbm::weightUpdate(arma::mat const &v_states, arma::mat &dw, arma::mat &dbh, arma::mat &dbv) { - arma::mat h_probs = prob(v_to_h(v_states)); arma::mat v_probs(dbv.n_rows, dbv.n_cols); - arma::mat h_states = h_probs; + arma::mat h_states = v_to_h(v_states); + arma::mat h_probs = prob(h_states); // Sample hidden - if (!m_params.doRaoBlackwell) + if (m_params.doRaoBlackwell) + { + h_states = h_probs; + } + else { h_states = sample(h_probs); }