Update hyperit.py

EdoardoChidichimo · Jun 24, 2024 · f1f382d · f1f382d
1 parent 99eca8f
commit f1f382d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/hyperit/hyperit.py b/hyperit/hyperit.py
@@ -447,6 +447,9 @@ def __which_estimator(self, measure: str) -> None:
                 self._Calc.setProperty(key, value)
 
         if initialise_parameter:
+            if self._measure == MeasureType.TE:
+                dataDim = self._data1.ndim
+                self._initialise_parameter = (dataDim, dataDim, *initialise_parameter)
             self._initialise_parameter = initialise_parameter
 
         if self._measure == MeasureType.TE and self._estimator_name == 'kernel':