- load and save of weights

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@588 b431acfa-c32f-4a4a-93f1-934dc6c82436
2019-10-28 17:26:59 +00:00
parent 27c07088c2
commit 9bfee1ae16
6 changed files with 114 additions and 22 deletions
@@ -32,8 +32,9 @@ public:

 	void train(size_t layerId, const arma::mat& batch, size_t miniBatchSize, size_t numEpochs, Rbm::IListener* pListener);
 	void train(const arma::mat& batch, size_t miniBatchSize, size_t numEpochs, Rbm::IListener* pListener);
-	void save(size_t numTraining);
-	void saveWeights();
+	bool save(size_t numTraining);
+	bool loadWeights();
+	bool saveWeights();
 	
 private:
 	const std::string &m_prjname;