jens/Rbm-legacy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[RBM]

Browse Source 
- miniBatchSize is parameter of train()

git-svn-id: http://moon:8086/svn/software/trunk/projects/RBM@307 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2016-07-07 23:34:03 +00:00
parent 0259727a2c
commit 2eabaa42ec
3 changed files with 15 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Source/Rbm.hpp
+1 -4
View File
@@ -37,7 +37,6 @@ public:
, m_doNormalizeData(false)
, m_doLearnVariance(false)
, m_numGibbs(1)
, m_miniBatchSize(100)
{
}
@@ -58,7 +57,6 @@ public:
bool m_doNormalizeData;
bool m_doLearnVariance;
size_t m_numGibbs;
size_t m_miniBatchSize;
};
Rbm(Weights &weights, const MatrixXd &batch);
@@ -78,7 +76,7 @@ public:
RowVectorXd calcMean(MatrixXd const &batch);
RowVectorXd calcSigma(MatrixXd const &batch);
MatrixXd calcZ(MatrixXd &v, MatrixXd &h);
void train(uint32_t numEpochs, double sigmaMin = 0.05);
void train(size_t numEpochs, size_t miniBatchSize, double sigmaMin = 0.05);
double getProgress() const;
double getEnergy(const VectorXd& visible, const VectorXd& hidden);
void toHidden(RowVectorXd &h, RowVectorXd const &v);
@@ -98,7 +96,6 @@ public:
void setNormalizeData(bool flag);
void setDoLearnVariance(bool flag);
void setNumGibbs(size_t value);
void setMiniBatchSize(size_t size);
void setMuWeights(double value);
void setMuSparsity(double value);
void setMomentum(double value);