New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
add progress bar to RBM fix #3889 #4228
add progress bar to RBM fix #3889 #4228
Conversation
looks good to me, lets wait for the CI. |
src/shogun/neuralnets/RBM.cpp
Outdated
@@ -136,6 +137,9 @@ void CRBM::train(CDenseFeatures<float64_t>* features) | |||
buffer = SGMatrix<float64_t>(m_num_hidden, m_batch_size); | |||
|
|||
int32_t counter = 0; | |||
|
|||
auto pb = progress(range(0, max_num_epochs)); | |||
|
|||
for (int32_t i=0; i<max_num_epochs; i++) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
You could have used the compact version here, instead of the pb.print_progress/complete
methods :) namely:
for (auto i : progress(range(0, max_num_epochs)))
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Ok, thanks. I will modify the code.
Ill merge once CI is green |
The CI fails again. How can I fix the error "The job exceeded the maximum time limit for jobs, and has been terminated" |
Unfortunately, that error is not something you can fix from your side because it is caused by Travis itself. However, this PR should be mergeable since all the other builds are green. |
Ok, thank you |
add progress bar to RBM algorithm