- added Stack::save()
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@579 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
+12
-4
@@ -21,20 +21,28 @@
|
||||
#include "Rbm.hpp"
|
||||
#include "ILayer.hpp"
|
||||
|
||||
class Layer : public ILayer
|
||||
class Layer
|
||||
{
|
||||
public:
|
||||
Layer *upper;
|
||||
Layer *lower;
|
||||
|
||||
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 saveWeights();
|
||||
|
||||
void up_pass(const arma::mat& v) override;
|
||||
void down_pass(arma::mat& dst, const arma::mat& src) override;
|
||||
void up_pass(const arma::mat& v);
|
||||
void down_pass(arma::mat& dst, const arma::mat& src);
|
||||
|
||||
size_t id()
|
||||
{
|
||||
return m_id;
|
||||
}
|
||||
|
||||
private:
|
||||
const std::string &m_prjname;
|
||||
std::string m_weightsFile;
|
||||
|
||||
Reference in New Issue
Block a user