- changed parameter order of train()

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

source/Rbm.cpp

@@ -38,7 +38,7 @@ Rbm::~Rbm()
 {
 }
 
-void Rbm::train(const arma::mat& batch, size_t numEpochs, size_t miniBatchSize, IListener* pListener)
+void Rbm::train(const arma::mat& batch, size_t miniBatchSize, size_t numEpochs, IListener* pListener)
 {
 	size_t epoch;
 	size_t gibbs;

source/Rbm.hpp

@@ -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);

source/main.cpp

@@ -120,7 +120,7 @@ int main()
 	arma::mat bv = arma::zeros(1, numVisible);
 	arma::mat bh = arma::zeros(1, numHidden);
 	Rbm rbm(params, w, bv, bh);
-	rbm.train(batch, 100, 100, &statusDisplay);
+	rbm.train(batch, 1000, 100, &statusDisplay);
 	saveWeight("mnist_2.weights.dat", 28, 28, w, bv, bh);
 	arma::mat v = arma::randu(numTraining, numVisible);
 	arma::mat h = rbm.toHiddenProbs(v);