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

- Stack holds training data

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@636 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2019-11-07 19:30:25 +00:00
parent 4c68cd4370
commit 3d6a6fbf00
4 changed files with 47 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/Stack.hpp
+10 -8
View File
@@ -44,23 +44,25 @@ public:
void delLayer(Layer *pLayer);
Layer* getLayer(size_t layerId) const;
void train(size_t layerId, const arma::mat& batch, Rbm::IListener* pListener);
void train(const arma::mat& batch, Rbm::IListener* pListener);
void train(Rbm::IListener* pListener);
bool load(LayerConstructor *pLayerConstructor=nullptr);
bool save();
void weightsInit(double stddev);
bool loadWeights();
bool saveWeights();
static size_t numTraining(const arma::mat &batch);
static void addTraining(arma::mat &batch, const arma::mat &toAdd);
static void delTraining(arma::mat &batch, int index);
arma::mat loadTraining();
void saveTraining(const arma::mat &batch);
size_t numTraining();
void addTraining(const arma::mat &toAdd);
void delTraining(int index);
size_t loadTraining();
size_t saveTraining();
arma::mat& trainingData();
arma::mat trainingData(Layer *pLayer);
private:
std::string m_name;
Layer *m_pLayers;
arma::mat m_trainingData;
};