- refactored

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@594 b431acfa-c32f-4a4a-93f1-934dc6c82436
2019-10-28 20:05:58 +00:00
parent 8a37ee6a69
commit 0f1631fe60
6 changed files with 34 additions and 34 deletions
@@ -0,0 +1,184 @@
+/*
+ * 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 "Layer.hpp"
+using namespace std;
+
+Layer::Layer(const string &name, size_t id, size_t numVisibleX, size_t numVisibleY, size_t numHidden)
+: Rbm(numVisibleX*numVisibleY, numHidden)
+, upper(nullptr)	
+, lower(nullptr)	
+, m_name(name)
+, m_id(id)
+, m_numVisibleX(numVisibleX)
+, m_numVisibleY(numVisibleY)
+{
+	cout << "Create Layer " << m_name << "." << to_string((int)m_id) << endl;
+	m_weightsFile = m_name + "." + to_string((int)m_id) + string(".weights.dat");
+}
+
+Layer::Layer(const Layer& orig)
+: Rbm(orig.bv().n_elem, orig.bh().n_elem)
+, upper(nullptr)	
+, lower(nullptr)	
+, m_name(orig.m_name)
+, m_weightsFile(orig.m_weightsFile)
+, m_id(orig.m_id)
+, m_numVisibleX(orig.m_numVisibleX)
+, m_numVisibleY(orig.m_numVisibleY)
+{
+}
+
+Layer::~Layer()
+{
+}
+
+bool Layer::loadWeights(const string &prjname)
+{
+	int numVisibleX;
+	int numVisibleY;
+	int numHidden;
+
+	string filename = m_weightsFile;
+	if (prjname.size() > 0)
+	{
+		filename = prjname + string(".") + m_weightsFile;
+	}
+
+	FILE *pFile = fopen(filename.c_str(),"r");
+	if (!pFile)
+	{
+		std::cout << "loadWeights(): Could not open " << filename << " for reading!" << std::endl;
+		return false;
+	}
+
+	int result = fscanf(pFile, "%d %d %d\n", &numVisibleX, &numVisibleY, &numHidden);
+	if (result < 0)
+	{
+		return false;
+	}
+
+	size_t numVisible = numVisibleX*numVisibleY;
+	
+	int i, j;
+	float v;
+	for (i=0; i < numVisible; i++)
+	{
+		result = fscanf(pFile, "%f", &v);
+		if (result > 0)
+		{
+			m_bv(i) = v;
+		}
+	}
+	for (i=0; i < numHidden; i++)
+	{
+		result = fscanf(pFile, "%f", &v);
+		if (result > 0)
+		{
+			m_bh(i) = v;
+		}
+	}
+	for (i=0; i < numVisible; i++)
+	{
+		for (j=0; j < numHidden; j++)
+		{
+
+			result = fscanf(pFile, "%f", &v);
+			if (result > 0)
+			{
+				m_w(i, j) = v;
+			}
+		}
+	}
+	fclose(pFile);
+	
+	return true;
+}
+	
+bool Layer::saveWeights(const string &prjname)
+{
+	string filename = m_weightsFile;
+	if (prjname.size() > 0)
+	{
+		filename = prjname + string(".") + m_weightsFile;
+	}
+
+	FILE *pFile = fopen(filename.c_str(),"w");
+	if (!pFile)
+	{
+		std::cout << "saveWeights(): Could not open " << filename << " for writing!" << std::endl;
+		return false;
+	}
+
+	int numHidden = m_bh.n_elem;
+	int numVisible = m_bv.n_elem;
+	fprintf(pFile, "%d %d %d\n", (int)m_numVisibleX, (int)m_numVisibleY, numHidden);
+
+	int i, j;
+
+	for (i=0; i < numVisible; i++)
+	{
+		fprintf(pFile, "%3.6f\n", m_bv(i));
+	}
+	for (i=0; i < numHidden; i++)
+	{
+		fprintf(pFile, "%3.6f\n", m_bh(i));
+	}
+	for (i=0; i < numVisible; i++)
+	{
+		for (j=0; j < numHidden; j++)
+		{
+			fprintf(pFile, "%3.6f ", m_w(i,j));
+		}
+		fprintf(pFile, "\n");
+	}
+	fclose(pFile);
+
+	return true;
+}
+
+Json::Value Layer::toJson() const
+{
+	std::cout << "Exporting Layer " << to_string((int)m_id) << std::endl;
+	Json::Value layer;
+	layer["name"] = m_name;
+	layer["id"] = (int)m_id;
+	layer["weights_file"] = m_weightsFile;
+	layer["numVisibleX"] = (int)m_numVisibleX;
+	layer["numVisibleY"] = (int)m_numVisibleY;
+	layer["numHidden"] = (int)m_bh.n_elem;
+	layer["rbm"] = Rbm::toJson();
+	
+	return layer;
+
+}
+
+arma::mat Layer::up_pass(const arma::mat &hidden)
+{
+	arma::mat reconstruction = toVisibleProbs(hidden);
+	if (upper)
+	{
+		return upper->up_pass(reconstruction);
+	}
+	return toHiddenProbs(reconstruction);
+}
+
+arma::mat Layer::down_pass(const arma::mat &visible)
+{
+	arma::mat hidden = toHiddenProbs(visible);
+	if (lower)
+	{
+		return lower->down_pass(hidden);
+	}
+	return toVisibleProbs(hidden);
+}