- moved all functions to Layer

- added Stack

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@577 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
2019-10-25 16:41:39 +00:00
parent 3e90330cd1
commit c68faa894e
7 changed files with 192 additions and 14 deletions
+12 -3
View File
@@ -19,19 +19,28 @@
#include <armadillo>
#include <jsoncpp/json/json.h>
#include "Rbm.hpp"
#include "ILayer.hpp"
class Layer
class Layer : public ILayer
{
public:
Layer(const std::string &prjname, size_t id, size_t numVisible, size_t numHidden);
Layer(const std::string &prjname, size_t id, size_t numVisibleX, size_t numVisibleY, size_t numHidden);
Layer(const Layer& orig);
virtual ~Layer();
Json::Value toJson() const;
Rbm& rbm();
void saveWeight();
void up_pass(const arma::mat& v) override;
void down_pass(arma::mat& dst, const arma::mat& src) override;
private:
const std::string &m_prjname;
std::string m_weightsFile;
size_t m_id;
size_t m_numVisibleX;
size_t m_numVisibleY;
Rbm::Params m_rbm_params;
Rbm m_rbm;