- added Stack::save()

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@579 b431acfa-c32f-4a4a-93f1-934dc6c82436
2019-10-25 17:30:31 +00:00
parent 5c01e7d4ce
commit 7aa3f0a747
6 changed files with 110 additions and 49 deletions
@@ -14,6 +14,8 @@
 #ifndef STACK_HPP
 #define STACK_HPP

+#include <string>
+#include <vector>
 #include <armadillo>
 #include <jsoncpp/json/json.h>
 #include "Layer.hpp"
@@ -21,15 +23,18 @@
 class Stack
 {
 public:
-	Stack();
+	Stack(const std::string &prjname);
 	Stack(const Stack& orig);
 	virtual ~Stack();
 	
-	void addLayer(const Layer &layer);
-	const Layer& getLayer();
+	void addLayer(Layer *pLayer);
+	const Layer* getLayer(size_t id);
+	void save(size_t numTraining);
 	
 private:
-
+	const std::string &m_prjname;
+	Layer *m_pLayers;
+	
 };

 #endif /* STACK_HPP */