- Layer: do gibbs sampling in calcContextBatch()
- RBM: added Gibbs sampler - Stack adjust training column vector according to needs git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@797 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
+2
-1
@@ -176,13 +176,14 @@ public:
|
||||
arma::mat diffErr_squared = diffErr % diffErr;
|
||||
return arma::sum(diffErr_squared, 1) * 1.0/diffErr_squared.n_cols;
|
||||
}
|
||||
void gibbs_hv(arma::mat &h_states, arma::mat &v_states);
|
||||
void gibbs_vh(arma::mat &v_probs, arma::mat &h_probs);
|
||||
|
||||
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);
|
||||
void gibbs(arma::mat &hv_states, arma::mat &v_states);
|
||||
|
||||
protected:
|
||||
arma::mat m_bhv;
|
||||
|
||||
Reference in New Issue
Block a user