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

- refactored

Browse Source 
- switch between cd_hinton hid/linear and cd_jens using doGaussionVisible (temporary solution)
This commit is contained in:
jens
2024-01-24 19:22:57 +01:00
parent a5ed0be991
commit 34b20f4fb3
2 changed files with 60 additions and 108 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/Rbm.hpp
+4 -1
View File
@@ -162,7 +162,10 @@ protected:
private:
arma::mat m_whv;
void contrastiveDivergence(arma::mat const &v_states, arma::mat &dwhv, arma::mat &dbhv, arma::mat &dbv);
void cd_hinton_hid_binary(arma::mat const &v_states, arma::mat &dwhv, arma::mat &dbhv, arma::mat &dbv);
void cd_hinton_hid_linear(arma::mat const &v_states, arma::mat &dwhv, arma::mat &dbhv, arma::mat &dbv);
void cd_jens(arma::mat const &v_states, arma::mat &dwhv, arma::mat &dbhv, arma::mat &dbv);
void cd(arma::mat const &v_states, arma::mat &dwhv, arma::mat &dbhv, arma::mat &dbv);
};