Skip to content
This repository was archived by the owner on Feb 6, 2024. It is now read-only.

Commit 0ef7817

Browse files
committed
Fix optimizeLinkedModel by switching stable BFGS
1 parent 032a219 commit 0ef7817

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

model/partitionmodel.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -347,7 +347,7 @@ double PartitionModel::optimizeLinkedModel(bool write_info, double gradient_epsi
347347
// lower_bound[i] = MIN_RATE*0.2;
348348
// upper_bound[i] = MAX_RATE*2.0;
349349
// }
350-
if (Params::getInstance().optimize_alg.find("NR") != string::npos)
350+
if (Params::getInstance().optimize_alg.find("BFGS-B") == string::npos)
351351
score = -minimizeMultiDimen(variables, ndim, lower_bound, upper_bound, bound_check, max(gradient_epsilon, TOL_RATE));
352352
else
353353
score = -L_BFGS_B(ndim, variables+1, lower_bound+1, upper_bound+1, max(gradient_epsilon, TOL_RATE));

0 commit comments

Comments
 (0)