- 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:
+12
-3
@@ -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;
|
||||
|
||||
|
||||
Reference in New Issue
Block a user