- crate and train Stack

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@586 b431acfa-c32f-4a4a-93f1-934dc6c82436
2019-10-26 09:49:12 +00:00
parent 9fd440e761
commit 91cc83d9ba
4 changed files with 63 additions and 9 deletions
@@ -28,8 +28,12 @@ public:
 	virtual ~Stack();
 	
 	void addLayer(RbmLayer *pLayer);
-	RbmLayer* getLayer(size_t id) const;
+	RbmLayer* getLayer(size_t layerId) const;
+
+	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();
 	
 private:
 	const std::string &m_prjname;
@@ -37,5 +41,6 @@ private:
 	
 };

+
 #endif /* STACK_HPP */