jens
f50607ef74
- create layers dynamically in main git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@583 b431acfa-c32f-4a4a-93f1-934dc6c82436
118 lines
2.6 KiB
C++
118 lines
2.6 KiB
C++
/*
|
|
* To change this license header, choose License Headers in Project Properties.
|
|
* To change this template file, choose Tools | Templates
|
|
* and open the template in the editor.
|
|
*/
|
|
|
|
/*
|
|
* File: Layer.cpp
|
|
* Author: jens
|
|
*
|
|
* Created on 25. Oktober 2019, 08:13
|
|
*/
|
|
|
|
#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 ¶ms)
|
|
: Rbm(params, numVisibleX*numVisibleY, numHidden)
|
|
, upper(nullptr)
|
|
, lower(nullptr)
|
|
, m_prjname(prjname)
|
|
, 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");
|
|
}
|
|
|
|
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_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)
|
|
{
|
|
}
|
|
|
|
RbmLayer::~RbmLayer()
|
|
{
|
|
}
|
|
|
|
void RbmLayer::saveWeights()
|
|
{
|
|
FILE *pFile = fopen(m_weightsFile.c_str(), "w");
|
|
|
|
if (!pFile)
|
|
{
|
|
std::cout << "Could not open " << m_weightsFile << "!" << std::endl;
|
|
return;
|
|
}
|
|
|
|
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));
|
|
}
|
|
for (i=0; i < numHidden; 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, "\n");
|
|
}
|
|
|
|
fclose(pFile);
|
|
|
|
}
|
|
|
|
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["weights_file"] = m_weightsFile;
|
|
layer["numVisibleX"] = to_string((int)m_numVisibleX);
|
|
layer["numVisibleY"] = to_string((int)m_numVisibleX);
|
|
layer["rbm"] = Rbm::toJson();
|
|
|
|
return layer;
|
|
|
|
}
|
|
|
|
arma::mat RbmLayer::up_pass(const arma::mat &hidden)
|
|
{
|
|
arma::mat reconstruction = toVisibleProbs(hidden);
|
|
if (upper)
|
|
{
|
|
return upper->up_pass(reconstruction);
|
|
}
|
|
return toHiddenProbs(reconstruction);
|
|
}
|
|
|
|
arma::mat RbmLayer::down_pass(const arma::mat &visible)
|
|
{
|
|
arma::mat hidden = toHiddenProbs(visible);
|
|
if (lower)
|
|
{
|
|
return lower->down_pass(hidden);
|
|
}
|
|
return toVisibleProbs(hidden);
|
|
}
|