From 84a5dd4550ddee86f59029e1b5055eb669104f78 Mon Sep 17 00:00:00 2001 From: jens Date: Wed, 24 Jan 2024 17:04:05 +0100 Subject: [PATCH] Refactored --- source/Rbm.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/source/Rbm.cpp b/source/Rbm.cpp index c6da7f8..ff039c5 100644 --- a/source/Rbm.cpp +++ b/source/Rbm.cpp @@ -76,10 +76,10 @@ arma::mat Rbm::toVisibleProbs(const arma::mat& hidden) const 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_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; }