- 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)
|
void saveProject(const string &prjname, const Layer &layer, size_t numTraining)
|
||||||
{
|
{
|
||||||
ofstream ofs(prjname + string(".prj"));
|
ofstream ofs(prjname + string(".prj"));
|
||||||
|
|||||||
Reference in New Issue
Block a user