diff --git a/source/Rbm.hpp b/source/Rbm.hpp index 2124815..9aa7279 100644 --- a/source/Rbm.hpp +++ b/source/Rbm.hpp @@ -142,10 +142,11 @@ public: private: Params m_params; arma::mat sample(arma::mat const &src); - void weightUpdate(arma::mat const &v_states, arma::mat &dw, arma::mat &dbh, arma::mat &dbv); - void weightUpdate(arma::mat const &v_states, arma::mat const &c_states, arma::mat &dwhv, arma::mat &dwhc, arma::mat &dbh, arma::mat &dbv, arma::mat &dbc); + void weightUpdate(arma::mat const &v_states, arma::mat &dw, arma::mat &dbh, arma::mat &dbv); + void weightUpdate(arma::mat const &v_states, arma::mat const &c_states, arma::mat &dwhv, arma::mat &dwhc, arma::mat &dbh, arma::mat &dbv, arma::mat &dbc); void uniform(arma::mat &srcDst, double stdDev=1.0, double mu=0.5); - void gibbs(arma::mat &h_states, arma::mat &v_states); + void gibbs(arma::mat &h_states, arma::mat &v_states); + protected: arma::mat m_whv; arma::mat m_whc;