Update GradientDescent.java

linqs · Sep 15, 2023 · 6c203f2 · 6c203f2
1 parent 92306a3
commit 6c203f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...ore/src/main/java/org/linqs/psl/application/learning/weight/gradient/GradientDescent.java b/...ore/src/main/java/org/linqs/psl/application/learning/weight/gradient/GradientDescent.java
@@ -551,7 +551,7 @@ protected void runValidationEvaluation(int epoch) {
         evaluation.compute(validationMap);
         currentValidationEvaluationMetric = evaluation.getNormalizedRepMetric();
 
-        if (currentValidationEvaluationMetric > bestValidationEvaluationMetric) {
+        if (MathUtils.compare(currentValidationEvaluationMetric, bestValidationEvaluationMetric) >= 0) {
             lastValidationImprovementEpoch = epoch;
 
             bestValidationEvaluationMetric = currentValidationEvaluationMetric;