- 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
2019-11-12 23:00:55 +00:00
parent c7ff22d940
commit 24f253e892
3 changed files with 9 additions and 3 deletions
@@ -36,7 +36,10 @@ public:

 	void train(arma::mat const &batch, IListener *pListener=nullptr)
 	{
-		Rbm::train(trainingData(batch), pListener);
+		if (batch.n_rows > 0)
+		{
+			Rbm::train(trainingData(batch), pListener);
+		}
 	}
 	
 	std::string& name()