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

- added Stack::loadTraining(), Stack::addTraing(), Stack::delTraining()

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@623 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2019-11-07 15:24:23 +00:00
parent a5dd0acc58
commit 5d281ddca6
3 changed files with 79 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/main.cpp
+9 -4
View File
@@ -84,10 +84,15 @@ int main()
RbmListener statusDisplay;
Stack stack(project);
arma::mat batch = loadTraining(project + string(".training.dat"));
arma::mat batch = stack.loadTraining();
size_t numTraining = batch.n_rows;
printf("Loaded %d training samples\n", (int)numTraining);
printf("Loaded %d training samples\n", (int)batch.n_rows);
stack.addTraining(batch, batch.row(1));
printf("Loaded %d training samples\n", (int)batch.n_rows);
stack.delTraining(batch, 0);
printf("Loaded %d training samples\n", (int)batch.n_rows);
#if 1
const int numLayers = 4;
@@ -128,7 +133,7 @@ int main()
stack.saveWeights();
Layer *layer = stack.getLayer(0);
arma::mat v = arma::randu(numTraining, layer->bv().n_elem);
arma::mat v = arma::randu(batch.n_rows, layer->bv().n_elem);
arma::mat h = layer->toHiddenProbs(v);
arma::mat r = layer->toVisibleProbs(h);
return 0;