jens/Rbm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

- refactored

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@752 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2022-01-08 10:56:36 +00:00
parent 637b237966
commit be9a4a4d74
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/Rbm.cpp
+7 -3
View File
@@ -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) 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 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 // Sample hidden
if (!m_params.doRaoBlackwell) if (m_params.doRaoBlackwell)
{
h_states = h_probs;
}
else
{ {
h_states = sample(h_probs); h_states = sample(h_probs);
} }