Fix shape issue in GlobalFrechet and bump version

matthieubulte · Jun 20, 2024 · 3e02448 · 3e02448
1 parent a848e3f
commit 3e02448
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyfrechet/regression/frechet_regression/global_frechet.py b/pyfrechet/regression/frechet_regression/global_frechet.py
@@ -17,5 +17,5 @@ def fit(self, X, y: MetricData):
         return self
 
     def weights_for(self, x):
-        S_inv_dx = cho_solve(self.Sigma_chol_, x - self.mu_)
+        S_inv_dx = cho_solve(self.Sigma_chol_, x.flatten() - self.mu_)
         return (1 + self.centered_x_train_ @ S_inv_dx)/self.centered_x_train_.shape[0]
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="pyfrechet",
-    version="1.0.2",
+    version="1.0.3",
     author="Matthieu Bulté",
     author_email="mb@math.ku.dk",
     description="A module for the manipulation and analysis of data in metric spaces.",