- added passes to layer

- removed redundancy


git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@784 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
2022-01-11 16:46:01 +00:00
parent d18b45ef50
commit eacefc1e80
3 changed files with 55 additions and 9 deletions
+6 -1
View File
@@ -84,13 +84,13 @@ int main()
stack.loadTrainingBatch();
#if CREATE_TEST
stack.addTraining(stack.trainingBatch().row(1));
printf("There are %d training samples\n", (int)stack.trainingBatch().n_rows);
stack.delTraining(0);
printf("There are %d training samples\n", (int)stack.trainingBatch().n_rows);
#if 1
const int numLayers = 4;
int i = 0;
Layer *lowerLayer = new Layer("Layer", i, 16, 16, 8);
@@ -132,5 +132,10 @@ int main()
arma::mat v = arma::randu(stack.trainingBatch().n_rows, layer->bv().n_elem);
arma::mat h = layer->toHiddenProbs(v);
arma::mat r = layer->toVisibleProbs(h);
v.print("v1");
r = layer->upDownPass(v);
r.print("v2");
return 0;
}