- add Rbm::normalize()

- normalize of training data git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@660 b431acfa-c32f-4a4a-93f1-934dc6c82436
2019-11-11 21:08:48 +00:00
parent 9593aea278
commit f610d2930e
5 changed files with 21 additions and 3 deletions
@@ -222,7 +222,7 @@ arma::mat Stack::trainingData(Layer* pThatLayer)
 	return thisBatch;
 }

-size_t Stack::loadTraining()
+size_t Stack::loadTraining(bool doNormalize)
 {
 	uint32_t numTraining = 0;
 	uint32_t numVisible = 0;
@@ -263,6 +263,11 @@ size_t Stack::loadTraining()
 	}
 	fclose(pFile);
 	std::cout << "Loaded " << numTraining << " training samples\n";
+	
+	if (doNormalize)
+	{
+		m_trainingData = Rbm::normalize(m_trainingData);
+	}
 	return numTraining;
 }