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

. improved toHidden(), toVisible()

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@564 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2019-10-22 06:20:43 +00:00
parent 8d323ee054
commit 675a5485fe
3 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/Rbm.cpp
+4 -4
View File
@@ -184,14 +184,14 @@ arma::mat Rbm::sample(const arma::mat &src)
arma::mat Rbm::toHidden(const arma::mat &visible)
{
arma::mat h = visible.t() * m_w + m_bh;
return h;
arma::mat h = visible * m_w + arma::repmat(m_bh, visible.n_rows, 1);
return probsLogistic(h);
}
arma::mat Rbm::toVisible(const arma::mat &hidden)
{
arma::mat v = hidden * m_w.t() + m_bv;
return v;
arma::mat v = hidden * m_w.t() + arma::repmat(m_bv, hidden.n_rows, 1);
return probsLogistic(v);
}
void Rbm::uniform(arma::mat& srcDst, double mu, double stdDev)