- moved Rbm:prob() to Matutils::prob()
- Matutils::Normalize uses prob()
This commit is contained in:
+2
-7
@@ -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);
|
||||
|
||||
Reference in New Issue
Block a user