diff --git a/fdasrsf/umap_metric.py b/fdasrsf/umap_metric.py index c7c4654..67f0beb 100644 --- a/fdasrsf/umap_metric.py +++ b/fdasrsf/umap_metric.py @@ -268,7 +268,9 @@ def warp_rbfgs(q1, q2): hCurGradNorm = norm2(hCurGradient) - lsstats = {"costevals": 0, "stepsize": 0.0, "alpha": 0.0} + costevals = 0 + stepsize = 0.0 + alpha = 0.0 ultimatum = False @@ -308,6 +310,9 @@ def warp_rbfgs(q1, q2): htilde = group_action_SRVF(htilde, hNext) q2tilde = group_action_SRVF(q2tilde, hNext) + stepsize = lsstats['stepsize'] + alpha = lsstats['alpha'] + alpha = stepsize / norm2(p) step = alpha * p