- added Stack::save()

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@579 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
2019-10-25 17:30:31 +00:00
parent 5c01e7d4ce
commit 7aa3f0a747
6 changed files with 110 additions and 49 deletions
+11 -9
View File
@@ -15,35 +15,37 @@
using namespace std;
Layer::Layer(const string &prjname, size_t id, size_t numVisibleX, size_t numVisibleY, size_t numHidden)
: m_prjname(prjname)
: upper(nullptr)
, lower(nullptr)
, m_prjname(prjname)
, m_id(id)
, m_numVisibleX(numVisibleX)
, m_numVisibleY(numVisibleY)
, m_rbm(m_rbm_params, numVisibleX*numVisibleY, numHidden)
{
m_weightsFile = m_prjname + string(".weights.") + to_string((int)m_id) + string(".dat");
}
Layer::Layer(const Layer& orig)
: m_prjname(orig.m_prjname)
: upper(nullptr)
, lower(nullptr)
, m_prjname(orig.m_prjname)
, m_weightsFile(orig.m_weightsFile)
, m_id(orig.m_id)
, m_numVisibleX(orig.m_numVisibleX)
, m_numVisibleY(orig.m_numVisibleY)
, m_rbm_params(orig.m_rbm_params)
, m_rbm(orig.m_rbm)
{
m_weightsFile = m_prjname + string(".weights.dat");
}
Layer::~Layer()
{
}
void Layer::saveWeight()
void Layer::saveWeights()
{
FILE *pFile;
pFile = fopen(m_weightsFile.c_str(), "w");
FILE *pFile = fopen(m_weightsFile.c_str(), "w");
if (!pFile)
{
@@ -91,7 +93,7 @@ Json::Value Layer::toJson() const
{
Json::Value layer;
layer["name"] = string("Layer ") + to_string((int)m_id);
layer["weights_file"] = m_prjname + string(".weights.dat");
layer["weights_file"] = m_weightsFile;
layer["numVisibleX"] = to_string((int)m_numVisibleX);
layer["numVisibleY"] = to_string((int)m_numVisibleX);
layer["rbm"] = m_rbm.toJson();