- fixed prototype for v_probs

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@755 b431acfa-c32f-4a4a-93f1-934dc6c82436
2022-01-08 11:16:20 +00:00
parent 37a973d0e4
commit 91bfa0e6ec
1 changed files with 6 additions and 1 deletions
@@ -108,9 +108,14 @@ void Rbm::gibbs(arma::mat &h_probs, arma::mat &v_probs)
 	}
 }

+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 Rbm::weightUpdate(arma::mat const &v_states, arma::mat &dw, arma::mat &dbh, arma::mat &dbv)
 {
-	arma::mat v_probs(dbv.n_rows, dbv.n_cols);
+	arma::mat v_probs(v_states);
 	arma::mat h_states = v_to_h(v_states);
 	arma::mat h_probs = prob(h_states);