Dev (#5)

* Minor optimizations

Author: sm00thix

cvmatrix/__init__.py

@@ -1 +1 @@
-__version__ = "2.0.1"
+__version__ = "2.0.2"

cvmatrix/cvmatrix.py

@@ -731,14 +731,14 @@ def _init_total_stats(self) -> None:
         ) and self.Y_total is not None:
             self.sum_Y_total = np.sum(self.Yw_total, axis=0, keepdims=True)
         if self.scale_X:
-            self.sum_sq_X_total = np.expand_dims(
-                np.einsum("ij, ij -> j", self.Xw_total, self.X_total), axis=0
+            self.sum_sq_X_total = np.sum(
+                self.Xw_total * self.X_total, axis=0, keepdims=True
             )
         else:
             self.sum_sq_X_total = None
         if self.scale_Y and self.Y_total is not None:
-            self.sum_sq_Y_total = np.expand_dims(
-                np.einsum("ij, ij -> j", self.Yw_total, self.Y_total), axis=0
+            self.sum_sq_Y_total = np.sum(
+                self.Yw_total * self.Y_total, axis=0, keepdims=True
             )
         else:
             self.sum_sq_Y_total = None

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cvmatrix"
-version = "2.0.1"
+version = "2.0.2"
 description = "Fast computation of possibly weighted and possibly centered/scaled training set kernel matrices in a cross-validation setting."
 authors = ["Sm00thix <oleemail@icloud.com>"]
 maintainers = ["Sm00thix <oleemail@icloud.com>"]