- refactored in gibbs

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@753 b431acfa-c32f-4a4a-93f1-934dc6c82436
2022-01-08 11:10:26 +00:00
parent be9a4a4d74
commit ef82ae7ea6
2 changed files with 41 additions and 28 deletions
@@ -142,9 +142,10 @@ public:
 private:
 	Params m_params;
 	arma::mat sample(arma::mat const &src);
-        void weightUpdate(arma::mat const &v_state, arma::mat &dw, arma::mat &dbh, arma::mat &dbv);
+        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);
 protected:
 	arma::mat m_whv;
 	arma::mat m_whc;