add progress bar to RBM fix #3889 (#4228)

shogun-toolbox · Apr 4, 2018 · 8f8347a · 8f8347a
1 parent 8d578f4
commit 8f8347a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/shogun/neuralnets/RBM.cpp b/src/shogun/neuralnets/RBM.cpp
@@ -35,6 +35,7 @@
 
 
 #include <shogun/base/Parameter.h>
+#include <shogun/base/progress.h>
 #include <shogun/mathematics/Math.h>
 #include <shogun/mathematics/eigen3.h>
 
@@ -136,7 +137,8 @@ void CRBM::train(CDenseFeatures<float64_t>* features)
 		buffer = SGMatrix<float64_t>(m_num_hidden, m_batch_size);
 
 	int32_t counter = 0;
-	for (int32_t i=0; i<max_num_epochs; i++)
+
+	for (auto i : progress(range(0, max_num_epochs)))
 	{
 		for (int32_t j=0; j < training_set_size; j += gd_mini_batch_size)
 		{