- added early stop of training

- fixed non-visible training data after load git-svn-id: http://moon:8086/svn/software/trunk/projects/RBM@554 b431acfa-c32f-4a4a-93f1-934dc6c82436
2019-10-17 19:48:14 +00:00
parent 0f2fe2b315
commit 2c2039ec2f
4 changed files with 25 additions and 7 deletions
@@ -48,7 +48,7 @@ public:
 	void sampleGaussian(MatrixXd &dst, MatrixXd const &src);
 	void sampleGaussian(MatrixXd &srcDst);
 	static MatrixXd normalizeData(MatrixXd const &src);
-	void train(size_t numEpochs, size_t miniBatchSize, double sigmaMin = 0.05);
+	void train(size_t numEpochs, size_t miniBatchSize, bool &doStop);
 	double getProgress() const;
 	void toHidden(RowVectorXd &h, RowVectorXd const &v);
 	void toVisible(RowVectorXd &v, RowVectorXd const &h);