Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ok

Browse Source
This commit is contained in:
tqchen 2015-03-21 00:37:59 -07:00
parent 14477f9f5a
commit 62330505e1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
rabit-learn/solver/lbfgs.h
View File

@ -177,7 +177,7 @@ class LBFGSSolver {
// swap new weight // swap new weight
std::swap(g.weight, g.grad); std::swap(g.weight, g.grad);
// check stop condition // check stop condition
if (gstate.num_iteration > min_lbfgs_iter) { if (gstate.num_iteration > static_cast<int>(min_lbfgs_iter)) {
if (g.old_objval - g.new_objval < lbfgs_stop_tol * g.init_objval) { if (g.old_objval - g.new_objval < lbfgs_stop_tol * g.init_objval) {
return true; return true;
} }
@ -196,7 +196,7 @@ class LBFGSSolver {
/*! \brief run optimization */ /*! \brief run optimization */
virtual void Run(void) { virtual void Run(void) {
this->Init(); this->Init();
while (gstate.num_iteration < max_lbfgs_iter) { while (gstate.num_iteration < static_cast<size_t>(max_lbfgs_iter)) {
if (this->UpdateOneIter()) break; if (this->UpdateOneIter()) break;
} }
if (silent == 0 && rabit::GetRank() == 0) { if (silent == 0 && rabit::GetRank() == 0) {