- changed parameter order of train()

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@571 b431acfa-c32f-4a4a-93f1-934dc6c82436
2019-10-24 18:54:40 +00:00
parent d6ed27f5d2
commit 0e544c1afc
3 changed files with 3 additions and 3 deletions
@@ -74,7 +74,7 @@ public:
 	Rbm(const Rbm& orig);
 	virtual ~Rbm();

-	void train(arma::mat const &batch, size_t numEpochs, size_t sizeMiniBatch, IListener *pListener);
+	void train(arma::mat const &batch, size_t miniBatchSize, size_t numEpochs, IListener *pListener);
 	
 	arma::mat toHiddenState(const arma::mat &visible);
 	arma::mat toVisibleState(const arma::mat &hidden);