jens/Rbm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

- train whv and whc separately

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@758 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2022-01-08 13:30:26 +00:00
parent b46e53f207
commit 3272899888
2 changed files with 85 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/Rbm.hpp
+4 -3
View File
@@ -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_hv(arma::mat const &v_states, arma::mat &dwhv, arma::mat &dbhv, arma::mat &dbv);
void weightUpdate_hc(arma::mat const &c_states, arma::mat &dwhc, arma::mat &dbhc, 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_hv(arma::mat &hv_states, arma::mat &v_states);
void gibbs_hc(arma::mat &hc_states, arma::mat &c_states);
protected:
arma::mat m_whv;