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

- refactored

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@594 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2019-10-28 20:05:58 +00:00
parent 8a37ee6a69
commit 0f1631fe60
6 changed files with 34 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/main.cpp
+4 -4
View File
@@ -7,7 +7,7 @@
#include <armadillo>
#include <jsoncpp/json/json.h>
#include "Rbm.hpp"
#include "RbmLayer.hpp"
#include "Layer.hpp"
#include "Stack.hpp"
using namespace std;
@@ -95,14 +95,14 @@ int main()
size_t numHidden = 256;
int i=0;
RbmLayer *lowerLayer = new RbmLayer("Layer", i, numVisibleX, numVisibleY, numHidden);
Layer *lowerLayer = new Layer("Layer", i, numVisibleX, numVisibleY, numHidden);
stack.addLayer(lowerLayer);
numHidden >>= 1;
i++;
for (i; i < 1; i++)
{
RbmLayer *layer = new RbmLayer("Layer", i, lowerLayer->bh().n_elem, 1, numHidden);
Layer *layer = new Layer("Layer", i, lowerLayer->bh().n_elem, 1, numHidden);
lowerLayer = layer;
stack.addLayer(layer);
numHidden >>= 1;
@@ -123,7 +123,7 @@ int main()
// Save weights
stack.saveWeights();
RbmLayer *layer = stack.getLayer(0);
Layer *layer = stack.getLayer(0);
arma::mat v = arma::randu(numTraining, layer->bv().n_elem);
arma::mat h = layer->toHiddenProbs(v);
arma::mat r = layer->toVisibleProbs(h);