diff --git a/source/Rbm.hpp b/source/Rbm.hpp index 6b5821e..3639cd5 100644 --- a/source/Rbm.hpp +++ b/source/Rbm.hpp @@ -136,19 +136,18 @@ public: arma::mat toVisibleProbs(const arma::mat &hidden) const; size_t numHidden() const; - size_t numVisible() const; arma::mat v_to_h(const arma::mat &visible) const; arma::mat h_to_v(const arma::mat &hidden) const; static arma::mat prob(arma::mat const &src); - static double rms_error_accu(arma::mat diffErr); - static arma::mat rms_error(arma::mat diffErr); - void gibbs_hv(arma::mat &h_states, arma::mat &v_states); void gibbs_vh(arma::mat &v_probs, arma::mat &h_probs); + static double rms_error_accu(arma::mat diffErr); + static arma::mat rms_error(arma::mat diffErr); + private: Params m_params; arma::mat sample(arma::mat const &src);