[RBM]
- added DBN stack - concentrated RBM params into structure git-svn-id: http://moon:8086/svn/software/trunk/projects/RBM@295 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
+22
-2
@@ -26,6 +26,16 @@
|
||||
#include "Rbm.hpp"
|
||||
//[/Headers]
|
||||
|
||||
|
||||
class IRbm
|
||||
{
|
||||
public:
|
||||
IRbm() {}
|
||||
virtual ~IRbm()
|
||||
{
|
||||
}
|
||||
};
|
||||
|
||||
class RbmComponentListener
|
||||
{
|
||||
public:
|
||||
@@ -84,19 +94,29 @@ public:
|
||||
void setMuWeights(double value);
|
||||
void setMuSparsity(double value);
|
||||
void setMomentum(double value);
|
||||
void addFromTraining();
|
||||
void batchchanged();
|
||||
void setWeightsIndex(size_t index);
|
||||
size_t getWeightsIndex();
|
||||
|
||||
void setTrainingIndex(size_t index);
|
||||
size_t getTrainingIndex();
|
||||
|
||||
void copyReconstructionToTraining();
|
||||
void redrawWeights();
|
||||
void redrawReconstruction();
|
||||
void redrawVariances();
|
||||
void train(size_t numEpochs);
|
||||
RowVectorXd const& getTrainingData();
|
||||
MatrixXd getHiddenBatch()
|
||||
MatrixXd const& getHiddenBatch()
|
||||
{
|
||||
return m_pRbm->getHiddenBatch();
|
||||
}
|
||||
|
||||
Rbm::Params const& params()
|
||||
{
|
||||
return m_pRbm->params();
|
||||
}
|
||||
|
||||
private:
|
||||
//[UserVariables] -- You can add your own custom variables in this section.
|
||||
ScopedPointer<Rbm> m_pRbm;
|
||||
|
||||
Reference in New Issue
Block a user