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

- improved test

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@787 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2022-01-11 19:13:47 +00:00
parent 5cb7b16d96
commit c6e8c89d39
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/main.cpp
+9 -3
View File
@@ -79,7 +79,6 @@ int main()
const string project("test"); const string project("test");
RbmListener statusDisplay;
Stack stack(".", project); Stack stack(".", project);
stack.loadTrainingBatch(); stack.loadTrainingBatch();
@@ -122,14 +121,17 @@ int main()
stack.loadWeights(); stack.loadWeights();
#endif #endif
#if TRAIN_TEST
RbmListener statusDisplay;
// Train stack // Train stack
stack.train(&statusDisplay); stack.train(&statusDisplay);
// Save weights // Save weights
stack.saveWeights(); stack.saveWeights();
#endif
Layer *layer = stack.getLayer(0); Layer *layer = stack.getLayer(0);
arma::mat v = arma::randu(stack.trainingBatch().n_rows, layer->bv().n_elem); arma::mat v = stack.trainingBatch();
arma::mat h = layer->toHiddenProbs(v); arma::mat h = layer->toHiddenProbs(v);
arma::mat r = layer->toVisibleProbs(h); arma::mat r = layer->toVisibleProbs(h);
@@ -137,5 +139,9 @@ int main()
r = layer->upDownPass(v); r = layer->upDownPass(v);
r.print("v2"); r.print("v2");
arma::mat err = Rbm::rms_error(r-v);
err.print("err");
return 0; return 0;
} }