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

- rbm: no training params at construction

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@591 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2019-10-28 19:20:34 +00:00
parent 711ad3057e
commit e2bd0ee850
6 changed files with 13 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/RbmLayer.cpp
+4 -6
View File
@@ -14,22 +14,21 @@
#include "RbmLayer.hpp"
using namespace std;
RbmLayer::RbmLayer(const string &name, size_t id, size_t numVisibleX, size_t numVisibleY, size_t numHidden, const Rbm::Params &params)
: Rbm(params, numVisibleX*numVisibleY, numHidden)
RbmLayer::RbmLayer(const string &name, size_t id, size_t numVisibleX, size_t numVisibleY, size_t numHidden)
: Rbm(numVisibleX*numVisibleY, numHidden)
, upper(nullptr)
, lower(nullptr)
, m_name(name)
, m_id(id)
, m_numVisibleX(numVisibleX)
, m_numVisibleY(numVisibleY)
, m_rbm_params(params)
{
cout << "Create Layer " << m_name << "." << to_string((int)m_id) << endl;
m_weightsFile = m_name + "." + to_string((int)m_id) + string(".weights.dat");
}
RbmLayer::RbmLayer(const RbmLayer& orig)
: Rbm(orig.m_rbm_params, orig.bv().n_elem, orig.bh().n_elem)
: Rbm(orig.bv().n_elem, orig.bh().n_elem)
, upper(nullptr)
, lower(nullptr)
, m_name(orig.m_name)
@@ -37,7 +36,6 @@ RbmLayer::RbmLayer(const RbmLayer& orig)
, m_id(orig.m_id)
, m_numVisibleX(orig.m_numVisibleX)
, m_numVisibleY(orig.m_numVisibleY)
, m_rbm_params(orig.m_rbm_params)
{
}
@@ -157,7 +155,7 @@ Json::Value RbmLayer::toJson() const
layer["id"] = (int)m_id;
layer["weights_file"] = m_weightsFile;
layer["numVisibleX"] = (int)m_numVisibleX;
layer["numVisibleY"] = (int)m_numVisibleX;
layer["numVisibleY"] = (int)m_numVisibleY;
layer["numHidden"] = (int)m_bh.n_elem;
layer["rbm"] = Rbm::toJson();