- load and save of weights

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@588 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
2019-10-28 17:26:59 +00:00
parent 27c07088c2
commit 9bfee1ae16
6 changed files with 114 additions and 22 deletions
+6 -3
View File
@@ -132,12 +132,15 @@ public:
private:
const Params &m_params;
arma::mat m_w;
arma::mat m_bh;
arma::mat m_bv;
arma::mat sample(arma::mat const &src);
static arma::mat probsLogistic(arma::mat const &src);
void uniform(arma::mat &srcDst, double mu=0.0, double stdDev=1.0);
protected:
arma::mat m_w;
arma::mat m_bh;
arma::mat m_bv;
};
#endif /* RBM_HPP */