Use stepsize not epsilon
@@ -89,7 +89,7 @@ public class NUTS {
double initialLogOfMasterP = getLogProb(probabilisticVertices);
- double stepSize = findStartingEpsilon(position,
+ double stepSize = findStartingStepSize(position,
gradient,
latentVertices,
probabilisticVertices,
@@ -627,7 +627,7 @@ public class NUTS {
}
}
- private static double findStartingEpsilon(Map<Long, DoubleTensor> position,
+ private static double findStartingStepSize(Map<Long, DoubleTensor> position,
Map<Long, DoubleTensor> gradient,
List<Vertex<DoubleTensor>> vertices,
List<Vertex> probabilisticVertices,
