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

Refactored

Browse Source
This commit is contained in:
jens
2024-01-24 17:04:05 +01:00
parent b467180535
commit 84a5dd4550
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/Rbm.cpp
+2 -2
View File
@@ -76,10 +76,10 @@ arma::mat Rbm::toVisibleProbs(const arma::mat& hidden) const
return Rbm::prob(h_to_v(hidden)); return Rbm::prob(h_to_v(hidden));
} }
void Rbm::weightsAssign(const arma::mat& w, const arma::mat& bhv, const arma::mat& bv) void Rbm::weightsAssign(const arma::mat& w, const arma::mat& bh, const arma::mat& bv)
{ {
m_whv.submat(0, 0, w.n_rows - 1, w.n_cols - 1) = w; m_whv.submat(0, 0, w.n_rows - 1, w.n_cols - 1) = w;
m_bh.submat(0, 0, bhv.n_rows - 1, bhv.n_cols - 1) = bhv; m_bh.submat(0, 0, bh.n_rows - 1, bh.n_cols - 1) = bh;
m_bv.submat(0, 0, bv.n_rows - 1, bv.n_cols - 1) = bv; m_bv.submat(0, 0, bv.n_rows - 1, bv.n_cols - 1) = bv;
} }