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

- refactored

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@820 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2022-01-17 15:14:55 +00:00
parent dbed14cefa
commit 370fdd7bbb
4 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/DeepStack.hpp
+6 -7
View File
@@ -29,7 +29,13 @@ public:
virtual ~DeepStack();
void train(Rbm::IListener* pListener);
void train(const arma::mat& batch, Rbm::IListener* pListener) override;
void train(size_t layerId, const arma::mat& batch, Rbm::IListener* pListener) override;
arma::mat upPass(size_t layerId, arma::mat const &v);
arma::mat downPass(size_t layerId, arma::mat const &h);
arma::mat upDownPass(size_t layerId, arma::mat const &v);
size_t numTraining();
void addTraining(const arma::mat &toAdd);
void delTraining(int index);
@@ -38,13 +44,6 @@ public:
arma::mat& trainingBatch();
arma::mat trainingBatch(Layer *pLayer);
arma::mat upPass(size_t layerId, arma::mat const &v);
arma::mat downPass(size_t layerId, arma::mat const &h);
arma::mat upDownPass(size_t layerId, arma::mat const &v);
void train(const arma::mat& batch, Rbm::IListener* pListener) override;
void train(size_t layerId, const arma::mat& batch, Rbm::IListener* pListener) override;
private:
arma::mat m_trainingBatch;
};