- 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:
2022-01-08 15:24:07 +00:00
parent 3272899888
commit 9fff2082d6
5 changed files with 48 additions and 127 deletions
+5 -2
View File
@@ -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");
}