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

- refactored common functions into matutils

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@855 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2022-01-21 07:46:40 +00:00
parent c10d9c6e38
commit aef3049856
4 changed files with 80 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/Rbm.hpp
+2 -7
View File
@@ -124,7 +124,6 @@ public:
void train(arma::mat const &batch, IListener *pListener=nullptr);
static arma::mat normalize(const arma::mat &hidden);
const arma::mat& whv() const;
const arma::mat& bv() const;
const arma::mat& bh() const;
@@ -148,18 +147,14 @@ public:
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);
void contrastiveDivergence(arma::mat const &v_states, arma::mat &dwhv, arma::mat &dbhv, arma::mat &dbv);
void uniform(arma::mat &srcDst, double stdDev=1.0, double mu=0.5);
protected:
Params m_params;
arma::mat m_bhv;
arma::mat m_bv;
private:
arma::mat m_whv;
void contrastiveDivergence(arma::mat const &v_states, arma::mat &dwhv, arma::mat &dbhv, arma::mat &dbv);
};