From 1f960761239dd26dd18afba7af2d68d08bed6312 Mon Sep 17 00:00:00 2001 From: Yi Zhang Date: Wed, 21 Oct 2020 10:49:26 +0800 Subject: [PATCH] update docs --- src/lib/sgd.h | 4 ++-- src/sample/sample.cpp | 6 +----- 2 files changed, 3 insertions(+), 7 deletions(-) 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; }