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

- crate and train Stack

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@586 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2019-10-26 09:49:12 +00:00
parent 9fd440e761
commit 91cc83d9ba
4 changed files with 63 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/Stack.cpp
+41 -2
View File
@@ -50,12 +50,12 @@ void Stack::addLayer(RbmLayer *pOtherLayer)
}
}
RbmLayer* Stack::getLayer(size_t id) const
RbmLayer* Stack::getLayer(size_t layerId) const
{
RbmLayer *pLayer = m_pLayers;
while(pLayer)
{
if (pLayer->id() == id)
if (pLayer->id() == layerId)
{
return pLayer;
}
@@ -88,4 +88,43 @@ void Stack::save(size_t numTraining)
ofs << writer.write(project);
}
void Stack::saveWeights()
{
RbmLayer *pLayer = m_pLayers;
while(pLayer)
{
pLayer->saveWeights();
}
}
void Stack::train(const arma::mat& batch, size_t miniBatchSize, size_t numEpochs, Rbm::IListener* pListener)
{
RbmLayer *pLayer = m_pLayers;
while(pLayer)
{
train(pLayer->id(), batch, miniBatchSize, numEpochs, pListener);
pLayer = pLayer->upper;
}
}
void Stack::train(size_t layerId, const arma::mat& batch, size_t miniBatchSize, size_t numEpochs, Rbm::IListener* pListener)
{
arma::mat thisBatch = batch;
RbmLayer *pLayer = m_pLayers;
while(pLayer)
{
if (pLayer->id() == layerId)
{
break;
}
thisBatch = pLayer->toHiddenProbs(thisBatch);
pLayer = pLayer->upper;
}
if (pLayer)
{
std::cout << m_prjname << ": " << " Training of layer " << std::to_string(layerId) << std::endl;
pLayer->train(thisBatch, miniBatchSize, numEpochs, pListener);
}
}