jens/Rbm-legacy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[RBM]

Browse Source 
- refactored and cleaned up

git-svn-id: http://moon:8086/svn/software/trunk/projects/RBM@290 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2016-06-15 20:32:28 +00:00
parent 3d56e4a163
commit 2de8e9ac48
7 changed files with 277 additions and 277 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Source/RbmComponent.h
+9 -4
View File
@@ -106,7 +106,9 @@ public:
void addFromTraining()
{
DrawReconstruction->setData(DrawTraining->getData());
DrawReconstruction->getData() = DrawTraining->getData();
DrawReconstruction->DrawData();
m_layers.add(DrawTraining->getData(), m_vNumX*m_vNumY);
m_listener.onLayerSizeChanged(m_layers.getSize());
}
@@ -124,14 +126,17 @@ public:
m_currTrainingIndexToDraw = index;
RowVectorXd t = m_layers.getAt(index);
DrawTraining->setData(t);
DrawTraining->getData() = t;
DrawTraining->DrawData();
redrawReconstruction();
}
}
void copyReconstructionToTraining()
{
DrawTraining->setData(DrawReconstruction->getData());
DrawTraining->getData() = DrawReconstruction->getData();
DrawTraining->DrawData();
}
void clearTraining()
@@ -159,7 +164,7 @@ public:
void redrawReconstruction();
void train(size_t numEpochs)
{
m_pRbm->train(m_layers, numEpochs);
m_pRbm->train(numEpochs);
}
private: