- refactored

source/DeepStack.cpp

@@ -69,10 +69,26 @@ void DeepStack::train(size_t layerId, const arma::mat& batch, Rbm::IListener* pL
 }
 
 arma::mat DeepStack::upPass(size_t layerId, const arma::mat& v)
+{
+	return upPass(getLayer(layerId), v);
+}
+
+arma::mat DeepStack::downPass(size_t layerId, const arma::mat& h)
+{
+	return downPass(getLayer(layerId), h);
+}
+
+arma::mat DeepStack::upDownPass(size_t layerId, const arma::mat& v)
+{
+	arma::mat h = upPass(layerId, v);
+	arma::mat r = downPass(numLayers()-1, h);
+	return r;
+}
+
+arma::mat DeepStack::upPass(Layer *pLayer, arma::mat const &v)
 {
 	arma::mat h = arma::zeros(0,0);
 	arma::mat tv = v;
-	Layer *pLayer = getLayer(layerId);
 	while(pLayer)
 	{
 		if (pLayer->isEnable())
@@ -85,11 +101,10 @@ arma::mat DeepStack::upPass(size_t layerId, const arma::mat& v)
 	return h;
 }
 
-arma::mat DeepStack::downPass(size_t layerId, const arma::mat& h)
+arma::mat DeepStack::downPass(Layer *pLayer, arma::mat const &h)
 {
 	arma::mat v = arma::zeros(0,0);
 	arma::mat th = h;
-	Layer *pLayer = getLayer(layerId);
 	while(pLayer)
 	{
 		if (pLayer->isEnable())
@@ -101,11 +116,3 @@ arma::mat DeepStack::downPass(size_t layerId, const arma::mat& h)
 	}
 	return v;
 }
-
-arma::mat DeepStack::upDownPass(size_t layerId, const arma::mat& v)
-{
-	arma::mat h = upPass(layerId, v);
-	arma::mat r = downPass(numLayers()-1, h);
-	return r;
-}
-

source/DeepStack.hpp

@@ -36,6 +36,9 @@ public:
     arma::mat downPass(size_t layerId, arma::mat const &h);
     arma::mat upDownPass(size_t layerId, arma::mat const &v);
 
+    arma::mat upPass(Layer *pLayer, arma::mat const &v);
+    arma::mat downPass(Layer *pLayer, arma::mat const &h);
+
 };