- weightInit are construction time parameter

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@593 b431acfa-c32f-4a4a-93f1-934dc6c82436
2019-10-28 20:04:39 +00:00
parent 43dc764daa
commit 8a37ee6a69
5 changed files with 14 additions and 85 deletions
@@ -13,13 +13,14 @@

 #include "Rbm.hpp"

-Rbm::Rbm(size_t numVisible, size_t numHidden)
+Rbm::Rbm(size_t numVisible, size_t numHidden, double weightInit)
 : m_params()
+, m_weightInit(weightInit)
 , m_w(numVisible, numHidden)
 , m_bv(1, numVisible)
 , m_bh(1, numHidden)
 {
-	weightsInit(0.0, m_params.weightInit);
+	weightsInit(0.0, m_weightInit);
 }

 Rbm::Rbm(const Rbm& orig)
@@ -52,11 +53,11 @@ Json::Value Rbm::toJson() const
 	Json::Value rbm;
 	rbm["numVisible"] = m_bv.n_elem;
 	rbm["numHidden"] = m_bh.n_elem;
+	rbm["weightInit"] = m_weightInit;
 	rbm["params"] = m_params.toJson();
 	return rbm;
 }

-
 void Rbm::train(const arma::mat& batch, size_t miniBatchSize, size_t numEpochs, IListener* pListener)
 {
 	Status status;