- 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:
+6
-1
@@ -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;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user