- refactored

- use Armadillo for load/save of weight and training data

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@775 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
2022-01-10 15:25:52 +00:00
parent 952cf26930
commit ff2086a1ff
9 changed files with 125 additions and 65 deletions
+5 -5
View File
@@ -98,12 +98,12 @@ private:
void clearTraining()
{
patterSlider->setRange(0, 0, 1);
m_stack->trainingData().clear();
m_stack->trainingBatch().clear();
}
void loadTraining()
{
m_stack->loadTraining(rbmNormalizeDataToggleButton->getToggleState());
m_stack->loadTrainingBatch(rbmNormalizeDataToggleButton->getToggleState());
patterSlider->setRange(0, m_stack->numTraining()-1, 1);
}
@@ -128,16 +128,16 @@ private:
{
if (m_pLayer->context().is_empty())
{
m_pLayer->setBatch(m_stack->trainingData());
m_pLayer->setBatch(m_stack->trainingBatch());
}
if (!m_pLayer->context().is_empty())
{
return arma::join_rows(m_stack->trainingData().row(index), m_pLayer->context().row(index));
return arma::join_rows(m_stack->trainingBatch().row(index), m_pLayer->context().row(index));
}
else
{
return m_stack->trainingData().row(index);
return m_stack->trainingBatch().row(index);
}
}