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

[RBM]

Browse Source 
- Layer is an Rbm

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@581 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2019-10-26 05:59:20 +00:00
parent 0463a64a93
commit eed6956971
3 changed files with 28 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/Layer.cpp
+16 -24
View File
@@ -14,20 +14,22 @@
#include "Layer.hpp"
using namespace std;
Layer::Layer(const string &prjname, size_t id, size_t numVisibleX, size_t numVisibleY, size_t numHidden)
: upper(nullptr)
Layer::Layer(const string &prjname, 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_id(id)
, m_numVisibleX(numVisibleX)
, m_numVisibleY(numVisibleY)
, m_rbm(m_rbm_params, numVisibleX*numVisibleY, numHidden)
, m_rbm_params(params)
{
m_weightsFile = m_prjname + string(".weights.") + to_string((int)m_id) + string(".dat");
}
Layer::Layer(const Layer& orig)
: upper(nullptr)
: Rbm(orig.m_rbm_params, orig.bv().n_elem, orig.bh().n_elem)
, upper(nullptr)
, lower(nullptr)
, m_prjname(orig.m_prjname)
, m_weightsFile(orig.m_weightsFile)
@@ -35,7 +37,6 @@ Layer::Layer(const Layer& orig)
, m_numVisibleX(orig.m_numVisibleX)
, m_numVisibleY(orig.m_numVisibleY)
, m_rbm_params(orig.m_rbm_params)
, m_rbm(orig.m_rbm)
{
}
@@ -53,29 +54,25 @@ void Layer::saveWeights()
return;
}
const arma::mat &bv = rbm().bv();
const arma::mat &bh = rbm().bh();
const arma::mat &w = rbm().w();
size_t numHidden = bh.n_elem;
size_t numVisible = bv.n_elem;
size_t numHidden = bh().n_elem;
size_t numVisible = bh().n_elem;
fprintf(pFile, "%d %d %d\n", (int)m_numVisibleX, (int)m_numVisibleY, (int)numHidden);
uint32_t i, j;
for (i=0; i < numVisible; i++)
{
fprintf(pFile, "%3.6f\n", bv(i));
fprintf(pFile, "%3.6f\n", bv()(i));
}
for (i=0; i < numHidden; i++)
{
fprintf(pFile, "%3.6f\n", bh(i));
fprintf(pFile, "%3.6f\n", bh()(i));
}
for (i=0; i < numVisible; i++)
{
for (j=0; j < numHidden; j++)
{
fprintf(pFile, "%3.6f ", w(i,j));
fprintf(pFile, "%3.6f ", w()(i,j));
}
fprintf(pFile, "\n");
}
@@ -84,11 +81,6 @@ void Layer::saveWeights()
}
Rbm& Layer::rbm()
{
return m_rbm;
}
Json::Value Layer::toJson() const
{
Json::Value layer;
@@ -96,7 +88,7 @@ Json::Value Layer::toJson() const
layer["weights_file"] = m_weightsFile;
layer["numVisibleX"] = to_string((int)m_numVisibleX);
layer["numVisibleY"] = to_string((int)m_numVisibleX);
layer["rbm"] = m_rbm.toJson();
layer["rbm"] = Rbm::toJson();
return layer;
@@ -104,20 +96,20 @@ Json::Value Layer::toJson() const
arma::mat Layer::up_pass(const arma::mat &hidden)
{
arma::mat reconstruction = rbm().toVisibleProbs(hidden);
arma::mat reconstruction = toVisibleProbs(hidden);
if (upper)
{
return upper->up_pass(reconstruction);
}
return rbm().toHiddenProbs(reconstruction);
return toHiddenProbs(reconstruction);
}
arma::mat Layer::down_pass(const arma::mat &visible)
{
arma::mat hidden = rbm().toHiddenProbs(visible);
arma::mat hidden = toHiddenProbs(visible);
if (lower)
{
return lower->down_pass(hidden);
}
return rbm().toVisibleProbs(hidden);
return toVisibleProbs(hidden);
}