- moved Rbm:prob() to Matutils::prob()

- Matutils::Normalize uses prob()
This commit is contained in:
2024-01-29 13:04:13 +01:00
parent fe8142627d
commit c0445f3bf8
5 changed files with 13 additions and 13 deletions
+2 -7
View File
@@ -68,12 +68,12 @@ double Rbm::rms_error_accu(arma::mat diffErr)
arma::mat Rbm::toHiddenProbs(const arma::mat& visible) const
{
return Rbm::prob(v_to_h(visible));
return prob(v_to_h(visible));
}
arma::mat Rbm::toVisibleProbs(const arma::mat& hidden) const
{
return Rbm::prob(h_to_v(hidden));
return prob(h_to_v(hidden));
}
void Rbm::weightsAssign(const arma::mat& w, const arma::mat& bh, const arma::mat& bv)
@@ -371,11 +371,6 @@ void Rbm::train(arma::mat const &batch, IListener* pListener)
}
}
arma::mat Rbm::prob(const arma::mat &src)
{
return 1 / (1 + (arma::exp(-src)));
}
arma::mat Rbm::v_to_h(const arma::mat &visible) const
{
return visible * m_whv + arma::repmat(m_bh, visible.n_rows, 1);