Merge branch 'main' into NeSyEBMLearning

linqs · Feb 22, 2024 · d06243e · d06243e
2 parents 9aaa05a + f5bf9b9
commit d06243e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/...java/org/linqs/psl/application/learning/weight/gradient/minimizer/BinaryCrossEntropy.java b/...java/org/linqs/psl/application/learning/weight/gradient/minimizer/BinaryCrossEntropy.java
@@ -54,8 +54,8 @@ protected float computeSupervisedLoss() {
 
             int proxRuleIndex = rvAtomIndexToProxRuleIndex.get(atomIndex);
 
-            supervisedLoss += -1.0f * (observedAtom.getValue() * Math.log(Math.max(proxRuleObservedAtoms[proxRuleIndex].getValue(), MathUtils.EPSILON_FLOAT))
-                    + (1.0f - observedAtom.getValue()) * Math.log(Math.max(1.0f - proxRuleObservedAtoms[proxRuleIndex].getValue(), MathUtils.EPSILON_FLOAT)));
+            supervisedLoss += (float) (-1.0f * (observedAtom.getValue() * Math.log(Math.max(proxRuleObservedAtoms[proxRuleIndex].getValue(), MathUtils.EPSILON_FLOAT))
+                                + (1.0f - observedAtom.getValue()) * Math.log(Math.max(1.0f - proxRuleObservedAtoms[proxRuleIndex].getValue(), MathUtils.EPSILON_FLOAT))));
         }
 
         return supervisedLoss;

diff --git a/.../main/java/org/linqs/psl/application/learning/weight/gradient/minimizer/SquaredError.java b/.../main/java/org/linqs/psl/application/learning/weight/gradient/minimizer/SquaredError.java
@@ -51,7 +51,7 @@ protected float computeSupervisedLoss() {
                 continue;
             }
 
-            supervisedLoss += Math.pow(proxRuleObservedAtoms[rvAtomIndexToProxRuleIndex.get(atomIndex)].getValue() - observedAtom.getValue(), 2.0f);
+            supervisedLoss += (float) Math.pow(proxRuleObservedAtoms[rvAtomIndexToProxRuleIndex.get(atomIndex)].getValue() - observedAtom.getValue(), 2.0f);
         }
 
         return supervisedLoss;