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/main.cpp
+9 -10
View File
@@ -93,22 +93,21 @@ int main()
printf("Loaded %d training samples\n", (int)numTraining);
Layer layer(project, 0, numVisibleX, numVisibleY, numHidden);
Layer layer1(project, 1, numVisibleX, numVisibleY, numHidden);
Layer layer2(project, 2, numVisibleX, numVisibleY, numHidden);
Layer layer3(project, 3, numVisibleX, numVisibleY, numHidden);
Layer layer0(project, 0, numVisibleX, numVisibleY, numHidden, rbmParams);
Layer layer1(project, 1, numVisibleX, numVisibleY, numHidden, rbmParams);
Layer layer2(project, 2, numVisibleX, numVisibleY, numHidden, rbmParams);
Layer layer3(project, 3, numVisibleX, numVisibleY, numHidden, rbmParams);
Stack stack(project);
stack.addLayer(&layer);
stack.addLayer(&layer0);
stack.addLayer(&layer1);
stack.addLayer(&layer2);
stack.addLayer(&layer3);
stack.save(numTraining);
Rbm &rbm = layer.rbm();
rbm.train(batch, 1000, 100, &statusDisplay);
layer.saveWeights();
layer0.train(batch, 1000, 100, &statusDisplay);
layer0.saveWeights();
arma::mat v = arma::randu(numTraining, numVisibleX*numVisibleY);
arma::mat h = rbm.toHiddenProbs(v);
arma::mat r = rbm.toVisibleProbs(h);
arma::mat h = layer0.toHiddenProbs(v);
arma::mat r = layer0.toVisibleProbs(h);
return 0;
}