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

- prepared for context processing

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@757 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2022-01-08 11:52:01 +00:00
parent 1698ce0a1b
commit b46e53f207
4 changed files with 36 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/Layer.cpp
+5 -5
View File
@@ -15,7 +15,7 @@
using namespace std;
Layer::Layer(const string &name, size_t id, size_t numVisibleX, size_t numVisibleY, size_t numHidden)
: Rbm(numVisibleX*numVisibleY, numHidden)
: Rbm(numVisibleX*numVisibleY, numHidden, numHidden)
, next(nullptr)
, prev(nullptr)
, m_name(name)
@@ -86,7 +86,7 @@ bool Layer::loadWeights(const string &prjname)
result = fscanf(pFile, "%f", &v);
if (result > 0)
{
m_bh(i) = v;
m_bhv(i) = v;
}
}
for (i=0; i < numVisible; i++)
@@ -108,7 +108,7 @@ bool Layer::loadWeights(const string &prjname)
bool Layer::saveWeights(const string &prjname)
{
int numHidden = m_bh.n_elem;
int numHidden = m_bhv.n_elem;
int numVisible = m_bv.n_elem;
string filename = m_weightsFile;
if (prjname.size() > 0)
@@ -134,7 +134,7 @@ bool Layer::saveWeights(const string &prjname)
}
for (i=0; i < numHidden; i++)
{
fprintf(pFile, "%3.6f\n", m_bh(i));
fprintf(pFile, "%3.6f\n", m_bhv(i));
}
for (i=0; i < numVisible; i++)
{
@@ -158,7 +158,7 @@ Json::Value Layer::toJson() const
layer["weights_file"] = m_weightsFile;
layer["numVisibleX"] = (int)m_numVisibleX;
layer["numVisibleY"] = (int)m_numVisibleY;
layer["numHidden"] = (int)m_bh.n_elem;
layer["numHidden"] = (int)m_bhv.n_elem;
layer["rbm"] = Rbm::toJson();
return layer;