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

- added early stop of training

Browse Source 
- fixed non-visible training data after load

git-svn-id: http://moon:8086/svn/software/trunk/projects/RBM@554 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2019-10-17 19:48:14 +00:00
parent 0f2fe2b315
commit 2c2039ec2f
4 changed files with 25 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Source/MainComponent.cpp
+13 -5
View File
@@ -485,15 +485,15 @@ void MainComponent::buttonClicked (Button* buttonThatWasClicked)
if (buttonThatWasClicked == trainButton)
{
//[UserButtonCode_trainButton] -- add your button handler code here..
if (rbmNormalizeDataToggleButton->getToggleState() == true)
if (isThreadRunning())
{
m_trainingData = Rbm::normalizeData(m_layers.data());
m_doStop = true;
}
else
{
m_trainingData = m_layers.data();
m_doStop = false;
startThread();
}
startThread();
//[/UserButtonCode_trainButton]
}
else if (buttonThatWasClicked == addButton)
@@ -874,11 +874,19 @@ const juce::String& MainComponent::getBaseDir()
void MainComponent::run()
{
m_pRbmComponentCurr->train((size_t)numEpochslabel->getText().getIntValue(), (size_t)sizeMiniBatch->getText().getIntValue());
m_pRbmComponentCurr->train((size_t)numEpochslabel->getText().getIntValue(), (size_t)sizeMiniBatch->getText().getIntValue(), m_doStop);
}
void MainComponent::onChanged(const LayerArray &obj)
{
if (rbmNormalizeDataToggleButton->getToggleState() == true)
{
m_trainingData = Rbm::normalizeData(m_layers.data());
}
else
{
m_trainingData = m_layers.data();
}
patterSlider->setRange(0, obj.getSize()-1, 1);
}