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

- added projects and training data

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@592 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2019-10-28 19:27:20 +00:00
parent e2bd0ee850
commit 43dc764daa
5 changed files with 811790 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/main.cpp
+9 -9
View File
@@ -79,7 +79,7 @@ int main()
{
printf("Hallo, Welt!\n");
const string project("mnist_2");
const string project("test");
RbmListener statusDisplay;
Stack stack(project);
@@ -87,20 +87,20 @@ int main()
arma::mat batch = loadTraining(project + string(".training.dat"));
size_t numTraining = batch.n_rows;
size_t numVisibleX = 28;
size_t numVisibleY = 28;
size_t numHidden = 256;
printf("Loaded %d training samples\n", (int)numTraining);
#if 1
#if 0
size_t numVisibleX = 16;
size_t numVisibleY = 16;
size_t numHidden = 256;
int i=0;
RbmLayer *lowerLayer = new RbmLayer("Layer", i, numVisibleX, numVisibleY, numHidden);
stack.addLayer(lowerLayer);
numHidden >>= 1;
i++;
for (i; i < 4; i++)
for (i; i < 1; i++)
{
RbmLayer *layer = new RbmLayer("Layer", i, lowerLayer->bh().n_elem, 1, numHidden);
lowerLayer = layer;
@@ -118,13 +118,13 @@ int main()
stack.loadWeights();
// Train stack
stack.train(batch, 1000, 100, &statusDisplay);
stack.train(batch, 1000, 1000, &statusDisplay);
// Save weights
stack.saveWeights();
RbmLayer *layer = stack.getLayer(0);
arma::mat v = arma::randu(numTraining, numVisibleX*numVisibleY);
arma::mat v = arma::randu(numTraining, layer->bv().n_elem);
arma::mat h = layer->toHiddenProbs(v);
arma::mat r = layer->toVisibleProbs(h);
return 0;