- refactored

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@825 b431acfa-c32f-4a4a-93f1-934dc6c82436
2022-01-17 20:34:44 +00:00
parent 51e6eef12e
commit 9b2861e3d8
5 changed files with 81 additions and 84 deletions
@@ -68,13 +68,21 @@ public:
 	virtual void train(size_t layerId, const arma::mat& batch, Rbm::IListener* pListener) = 0;
 	virtual void train(const arma::mat& batch, Rbm::IListener* pListener) = 0;
 	arma::mat trainingBatchFrom(size_t layerId, const arma::mat& batch);	
-	
+
+	size_t numTraining();
+	void addTraining(const arma::mat &toAdd);
+	void delTraining(int index);
+	size_t loadTrainingBatch(const std::string &dir, bool doNormalize=false);
+	size_t saveTrainingBatch(const std::string &dir);
+	arma::mat& trainingBatch();
+
 protected:
 	StackType m_type;
 	std::string m_name;
 	Layer *m_pLayers;

 private:
+	arma::mat m_trainingBatch;
 	
 };