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

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@623 b431acfa-c32f-4a4a-93f1-934dc6c82436
2019-11-07 15:24:23 +00:00
parent a5dd0acc58
commit 5d281ddca6
3 changed files with 79 additions and 10 deletions
@@ -39,7 +39,9 @@ public:
 	Stack(const Stack& orig);
 	virtual ~Stack();
 	
+	size_t numLayers();
 	void addLayer(Layer *pLayer);
+	void delLayer(Layer *pLayer);
 	Layer* getLayer(size_t layerId) const;

 	void train(size_t layerId, const arma::mat& batch, Rbm::IListener* pListener);
@@ -50,12 +52,10 @@ public:
 	bool loadWeights();
 	bool saveWeights();
 	
-	void batchchanged()
-	{
-		std::cout << "Stack::batchchanged() called" << std::endl;
-	}
-	
-	size_t numLayers();
+	static size_t numTraining(const arma::mat &batch);
+	static void addTraining(arma::mat &batch, const arma::mat &toAdd);
+	static void delTraining(arma::mat &batch, int index);
+	arma::mat loadTraining();
 	
 private:
 	std::string m_name;