- fixed crash, when try to train without training patterns

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@664 b431acfa-c32f-4a4a-93f1-934dc6c82436

source/Layer.hpp

@@ -35,9 +35,12 @@ public:
 	bool saveWeights(const std::string &prjname="");
 
 	void train(arma::mat const &batch, IListener *pListener=nullptr)
+	{
+		if (batch.n_rows > 0)
 		{
 			Rbm::train(trainingData(batch), pListener);
 		}
+	}
 	
 	std::string& name()
 	{

source/MainComponent.cpp

@@ -762,7 +762,10 @@ void MainComponent::comboBoxChanged (ComboBox* comboBoxThatHasChanged)
 		m_pLayer = static_cast<RbmComponent*>(m_stack->getLayer(index));
 		updateControls();
 		m_pLayer->redrawWeights(m_weightIndex);
+		if (m_stack->trainingData().n_rows > 0)
+		{
 			m_pLayer->setTrainingData(m_stack->trainingData().row(m_trainingIndex));
+		}
         //[/UserComboBoxCode_m_rbmSelect]
     }
 

source/Stack.cpp

@@ -198,7 +198,7 @@ void Stack::train(Rbm::IListener* pListener)
 	while(pLayer)
 	{
 		std::cout << m_name << ": " << " Training of layer " << std::to_string(pLayer->id()) << std::endl;
-		pLayer->train(trainingData(pLayer), pListener);
+		pLayer->train(m_trainingData, pListener);
 		pLayer = pLayer->next;
 	}
 }