zhangyiss/libsgd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update docs

Browse Source
This commit is contained in:
张壹 2020-10-21 10:49:26 +08:00
parent 8df8c1c5e2
commit 1f96076123
2 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/lib/sgd.h
View File

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

6
src/sample/sample.cpp
View File

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