- added setBatch()

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@771 b431acfa-c32f-4a4a-93f1-934dc6c82436

source/Layer.hpp

@@ -33,8 +33,8 @@ public:
 	Json::Value toJson() const;
 	bool loadWeights(const std::string &prjname="");
 	bool saveWeights(const std::string &prjname="");
-
-	void train(arma::mat const &batch, IListener *pListener=nullptr)
+	
+	void setBatch(arma::mat const &batch)
 	{
 		if (batch.n_rows > 0)
 		{	
@@ -49,11 +49,11 @@ public:
 					c_states.row(i) = ctx;
 				}
 				arma::mat batch_with_ctx = arma::join_rows(batch, c_states);
-				Rbm::train(trainingData(batch_with_ctx), pListener);
+				Rbm::setBatch(trainingData(batch_with_ctx));
 			}
 			else
 			{
-				Rbm::train(trainingData(batch), pListener);
+				Rbm::setBatch(trainingData(batch));
 			}
 		}
 	}
@@ -136,6 +136,7 @@ private:
 	size_t m_numVisibleX;
 	size_t m_numVisibleY;
 	size_t m_numContext;
+	arma::mat m_context;
 
 };