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

- read/write of Stack project

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@589 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2019-10-28 18:55:12 +00:00
parent 9bfee1ae16
commit 43b475eac1
5 changed files with 71 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/RbmLayer.cpp
+26 -15
View File
@@ -14,25 +14,25 @@
#include "RbmLayer.hpp"
using namespace std;
RbmLayer::RbmLayer(const string &prjname, size_t id, size_t numVisibleX, size_t numVisibleY, size_t numHidden, const Rbm::Params &params)
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)
, upper(nullptr)
, lower(nullptr)
, m_prjname(prjname)
, m_name(name)
, m_id(id)
, m_numVisibleX(numVisibleX)
, m_numVisibleY(numVisibleY)
, m_rbm_params(params)
{
cout << "Create Layer " << m_prjname << "::" << to_string((int)m_id) << endl;
m_weightsFile = m_prjname + string(".weights.") + to_string((int)m_id) + string(".dat");
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)
, upper(nullptr)
, lower(nullptr)
, m_prjname(orig.m_prjname)
, m_name(orig.m_name)
, m_weightsFile(orig.m_weightsFile)
, m_id(orig.m_id)
, m_numVisibleX(orig.m_numVisibleX)
@@ -45,18 +45,22 @@ RbmLayer::~RbmLayer()
{
}
bool RbmLayer::loadWeights()
bool RbmLayer::loadWeights(const string &prjname)
{
int numVisibleX;
int numVisibleY;
int numHidden;
FILE *pFile;
pFile = fopen(m_weightsFile.c_str(),"r");
string filename = m_weightsFile;
if (prjname.size() > 0)
{
filename = prjname + string(".") + m_weightsFile;
}
FILE *pFile = fopen(filename.c_str(),"r");
if (!pFile)
{
std::cout << "loadWeights(): Could not open " << m_weightsFile << " for reading!" << std::endl;
std::cout << "loadWeights(): Could not open " << filename << " for reading!" << std::endl;
return false;
}
@@ -103,13 +107,18 @@ bool RbmLayer::loadWeights()
return true;
}
bool RbmLayer::saveWeights()
bool RbmLayer::saveWeights(const string &prjname)
{
FILE *pFile = fopen(m_weightsFile.c_str(), "w");
string filename = m_weightsFile;
if (prjname.size() > 0)
{
filename = prjname + string(".") + m_weightsFile;
}
FILE *pFile = fopen(filename.c_str(),"w");
if (!pFile)
{
std::cout << "saveWeights(): Could not open " << m_weightsFile << " for writing!" << std::endl;
std::cout << "saveWeights(): Could not open " << filename << " for writing!" << std::endl;
return false;
}
@@ -144,10 +153,12 @@ Json::Value RbmLayer::toJson() const
{
std::cout << "Exporting RbmLayer " << to_string((int)m_id) << std::endl;
Json::Value layer;
layer["name"] = string("Layer ") + to_string((int)m_id);
layer["name"] = m_name;
layer["id"] = (int)m_id;
layer["weights_file"] = m_weightsFile;
layer["numVisibleX"] = to_string((int)m_numVisibleX);
layer["numVisibleY"] = to_string((int)m_numVisibleX);
layer["numVisibleX"] = (int)m_numVisibleX;
layer["numVisibleY"] = (int)m_numVisibleX;
layer["numHidden"] = (int)m_bh.n_elem;
layer["rbm"] = Rbm::toJson();
return layer;