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@594 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2019-10-28 20:05:58 +00:00
parent 8a37ee6a69
commit 0f1631fe60
6 changed files with 34 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/Stack.hpp
+4 -4
View File
@@ -18,7 +18,7 @@
#include <vector>
#include <armadillo>
#include <jsoncpp/json/json.h>
#include "RbmLayer.hpp"
#include "Layer.hpp"
class Stack
{
@@ -27,8 +27,8 @@ public:
Stack(const Stack& orig);
virtual ~Stack();
void addLayer(RbmLayer *pLayer);
RbmLayer* getLayer(size_t layerId) const;
void addLayer(Layer *pLayer);
Layer* 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);
@@ -39,7 +39,7 @@ public:
private:
const std::string &m_prjname;
RbmLayer *m_pLayers;
Layer *m_pLayers;
};