jens/Rbm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

- added StackCreator

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@823 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2022-01-17 18:40:50 +00:00
parent 64ef87879f
commit e55f8e2153
12 changed files with 284 additions and 104 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/DeepStack.hpp
+5 -4
View File
@@ -24,8 +24,9 @@
class DeepStack : public AStack
{
public:
DeepStack(const std::string &dir, const std::string &name, StackType type=StackType::None);
DeepStack(const DeepStack& orig);
DeepStack(const std::string &name);
DeepStack(const std::string &name, Json::Value &layers, LayerConstructor *pLayerConstructor=nullptr);
DeepStack(const DeepStack& orig) = delete;
virtual ~DeepStack();
void train(const arma::mat& batch, Rbm::IListener* pListener) override;
@@ -38,8 +39,8 @@ public:
size_t numTraining();
void addTraining(const arma::mat &toAdd);
void delTraining(int index);
size_t loadTrainingBatch(bool doNormalize=false);
size_t saveTrainingBatch();
size_t loadTrainingBatch(const std::string &dir, bool doNormalize=false);
size_t saveTrainingBatch(const std::string &dir);
arma::mat& trainingBatch();
private: