diff --git a/src/lib/sgd.h b/src/lib/sgd.h index 4cb2f2a..06dcc46 100644 --- a/src/lib/sgd.h +++ b/src/lib/sgd.h @@ -103,8 +103,8 @@ typedef struct sgd_float epsilon; /** - * Damping rate of the classic momentum method, which is typically given - * between 0 and 1. The default is 0.01. + * Damping rate of the classic momentum method and the NAG method, which + * is typically given between 0 and 1. The default is 0.01. */ sgd_float mu; diff --git a/src/sample/sample.cpp b/src/sample/sample.cpp index 6486bc2..b036e38 100644 --- a/src/sample/sample.cpp +++ b/src/sample/sample.cpp @@ -78,11 +78,7 @@ int progress(void *instance, sgd_float fx, const sgd_float *x, const sgd_float * const sgd_para *param, const int n_size, const int k) { std::clog << "iteration time: " << k << ", fx: " << fx << "\r"; - if (fx < param->epsilon) - { - std::clog << std::endl; - std::cout << ">" << std::endl; - } + if (fx < param->epsilon) {std::clog << std::endl;} return 0; }