[RBM]
- removed classes HiddenLayer.hpp and VisibleLayer.hpp - fixed warnings - RbmComponent inherits Rbm - improved Rbm::train - use gaussion weight initialization git-svn-id: http://moon:8086/svn/software/trunk/projects/RBM@297 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
+4
-5
@@ -82,22 +82,21 @@ public:
|
||||
m_w.resize(m_numVisible, m_numHidden);
|
||||
m_bv.resize(m_numVisible);
|
||||
m_bh.resize(m_numHidden);
|
||||
shuffle(1.0);
|
||||
shuffle(0.01);
|
||||
}
|
||||
|
||||
void shuffle(double stdDev)
|
||||
{
|
||||
uint32_t i, j;
|
||||
double kdev = stdDev*sqrt(12.0);
|
||||
|
||||
m_bv.array().fill(Noise_Uniform(&m_noise));
|
||||
m_bh.array().fill(Noise_Uniform(&m_noise));
|
||||
m_bv.array().fill(stdDev*Noise_Gaussian(&m_noise));
|
||||
m_bh.array().fill(stdDev*Noise_Gaussian(&m_noise));
|
||||
|
||||
for (i=0; i < m_numVisible; i++)
|
||||
{
|
||||
for (j=0; j < m_numHidden; j++)
|
||||
{
|
||||
m_w(i,j) = kdev*Noise_Uniform(&m_noise);
|
||||
m_w(i,j) = stdDev*Noise_Gaussian(&m_noise);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user