- cleaned up
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@578 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
@@ -74,44 +74,6 @@ arma::mat loadTraining(const string &filename)
|
||||
|
||||
}
|
||||
|
||||
void saveWeight(const string &filename, size_t numVisibleX, size_t numVisibleY, const arma::mat &w, const arma::mat &bv, const arma::mat &bh)
|
||||
{
|
||||
FILE *pFile;
|
||||
|
||||
pFile = fopen(filename.c_str(), "w");
|
||||
|
||||
if (!pFile)
|
||||
{
|
||||
std::cout << "Could not open " << filename << "!" << std::endl;
|
||||
return;
|
||||
}
|
||||
|
||||
size_t numHidden = bh.n_elem;
|
||||
size_t numVisible = bv.n_elem;
|
||||
fprintf(pFile, "%d %d %d\n", (int)numVisibleX, (int)numVisibleY, (int)numHidden);
|
||||
|
||||
uint32_t i, j;
|
||||
|
||||
for (i=0; i < numVisible; i++)
|
||||
{
|
||||
fprintf(pFile, "%3.6f\n", bv(i));
|
||||
}
|
||||
for (i=0; i < numHidden; i++)
|
||||
{
|
||||
fprintf(pFile, "%3.6f\n", bh(i));
|
||||
}
|
||||
for (i=0; i < numVisible; i++)
|
||||
{
|
||||
for (j=0; j < numHidden; j++)
|
||||
{
|
||||
fprintf(pFile, "%3.6f ", w(i,j));
|
||||
}
|
||||
fprintf(pFile, "\n");
|
||||
}
|
||||
|
||||
fclose(pFile);
|
||||
}
|
||||
|
||||
void saveProject(const string &prjname, const Layer &layer, size_t numTraining)
|
||||
{
|
||||
ofstream ofs(prjname + string(".prj"));
|
||||
|
||||
Reference in New Issue
Block a user