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

- its gaussion hidden (not gaussian visible)

Browse Source
This commit is contained in:
jens
2024-01-24 15:14:30 +01:00
parent fe9b1951ea
commit 9a381c1899
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/Rbm.cpp
+3 -3
View File
@@ -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)