- rename Layer into RbmLayer

- add some log info on toJson() git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@582 b431acfa-c32f-4a4a-93f1-934dc6c82436
2019-10-26 06:07:13 +00:00
parent eed6956971
commit d16f0de8fa
8 changed files with 38 additions and 32 deletions
@@ -0,0 +1,116 @@
+/*
+ * 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 &params)
+: 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)
+{
+	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);
+}