- use all-at-onec (v+c) method
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@759 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
+5
-2
@@ -89,6 +89,7 @@ bool Layer::loadWeights(const string &prjname)
|
||||
m_bhv(i) = v;
|
||||
}
|
||||
}
|
||||
arma::mat weights(numVisible, numHidden);
|
||||
for (i=0; i < numVisible; i++)
|
||||
{
|
||||
for (j=0; j < numHidden; j++)
|
||||
@@ -97,10 +98,11 @@ bool Layer::loadWeights(const string &prjname)
|
||||
result = fscanf(pFile, "%f", &v);
|
||||
if (result > 0)
|
||||
{
|
||||
m_whv(i, j) = v;
|
||||
weights(i, j) = v;
|
||||
}
|
||||
}
|
||||
}
|
||||
weightsAssign(weights);
|
||||
fclose(pFile);
|
||||
|
||||
return true;
|
||||
@@ -136,11 +138,12 @@ bool Layer::saveWeights(const string &prjname)
|
||||
{
|
||||
fprintf(pFile, "%3.6f\n", m_bhv(i));
|
||||
}
|
||||
const arma::mat &_whv = whv();
|
||||
for (i=0; i < numVisible; i++)
|
||||
{
|
||||
for (j=0; j < numHidden; j++)
|
||||
{
|
||||
fprintf(pFile, "%3.6f ", m_whv(i,j));
|
||||
fprintf(pFile, "%3.6f ", _whv(i,j));
|
||||
}
|
||||
fprintf(pFile, "\n");
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user