- refactored

- use Armadillo for load/save of weight and training data git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@775 b431acfa-c32f-4a4a-93f1-934dc6c82436
2022-01-10 15:25:52 +00:00
parent 952cf26930
commit ff2086a1ff
9 changed files with 125 additions and 65 deletions
@@ -55,16 +55,16 @@ public:
 	size_t numTraining();
 	void addTraining(const arma::mat &toAdd);
 	void delTraining(int index);
-	size_t loadTraining(bool doNormalize=false);
-	size_t saveTraining();
-	arma::mat& trainingData();
-	arma::mat trainingData(Layer *pLayer);
+	size_t loadTrainingBatch(bool doNormalize=false);
+	size_t saveTrainingBatch();
+	arma::mat& trainingBatch();
+	arma::mat trainingBatch(Layer *pLayer);

 private:
 	std::string m_dir;
 	std::string m_name;
 	Layer *m_pLayers;
-	arma::mat m_trainingData;
+	arma::mat m_trainingBatch;
 	
 };