fix dimensionality init

settylab · Jul 11, 2024 · c7c1290 · c7c1290
1 parent df980d2
commit c7c1290
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,6 +1,7 @@
 # v1.4.4rc
 
  - remove `numpy` as direct dependency
+ - bugfix DimensionalityEstimator dimensionality initialization
 
 # v1.4.3
 

diff --git a/mellon/parameters.py b/mellon/parameters.py
@@ -12,6 +12,7 @@
     ndarray,
     ones,
     zeros,
+    full,
 )
 from jax.numpy import sum as arraysum
 from jax.numpy import any as arrayany
@@ -846,6 +847,8 @@ def compute_initial_dimensionalities(x, mu_dim, mu_dens, L, nn_distances, d):
     :rtype: array-like
     """
     target = log(d) - mu_dim
+    if asarray(target).size == 1:
+        target = full(L.shape[0], target)
     initial_dims = Ridge(fit_intercept=False).fit(L, target).coef_
     initial_dens = compute_initial_value(nn_distances, d, mu_dens, L)
     initial_value = stack([initial_dims, initial_dens])