- 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:
+11
-9
@@ -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();
|
||||
|
||||
Reference in New Issue
Block a user