supidly introduced bug after getting correct number..

beast-dev · Nov 21, 2023 · e272842 · e272842
1 parent 75001a6
commit e272842
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/dr/evomodel/coalescent/smooth/SkyGlideLikelihood.java b/src/dr/evomodel/coalescent/smooth/SkyGlideLikelihood.java
@@ -155,7 +155,7 @@ public double[] getGradientWrtLogPopulationSize() {
                     currentGridIndex = lastGridIndex;
                 }
             }
-//            updateSingleTreePopulationInverseGradientWrtLogPopSize(index, gradient);
+            updateSingleTreePopulationInverseGradientWrtLogPopSize(index, gradient);
         }
 
         return gradient;
@@ -198,7 +198,7 @@ public double getSingleTreeLogLikelihood(int index) {
                 lnL -= 0.5 * lineageCount * (lineageCount - 1) * sum;
             }
         }
-//        lnL += getSingleTreePopulationInverseLogLikelihood(index);
+        lnL += getSingleTreePopulationInverseLogLikelihood(index);
         return lnL;
     }
 
@@ -328,7 +328,7 @@ private void updateGridInterceptDerivativeWrtLogPopSize(int gridIndex, double[]
             final double lastGridTime = gridPointParameter.getParameterValue(gridIndex - 1);
 
             final double firstDerivative = thisGridTime / (thisGridTime - lastGridTime) * multiplier;
-            final double secondDerivative = -thisGridTime / (thisGridTime - lastGridTime) * multiplier;
+            final double secondDerivative = -lastGridTime / (thisGridTime - lastGridTime) * multiplier;
 
 //            if (logPopSizeParameter.getParameterValue(gridIndex) == logPopSizeParameter.getParameterValue(gridIndex + 1)) {
 //                gradient[gridIndex] += (firstDerivative + secondDerivative) / 2;