trying to have the basis matrix as the identity matrix

Author: Jimmy-INL

pysensors/basis/_base.py

@@ -43,9 +43,9 @@ def matrix_representation(self, n_basis_modes=None, copy=False):
         n_basis_modes = self._validate_input(n_basis_modes)
 
         if copy:
-            return self.basis_matrix_[:, :n_basis_modes].copy()
+            return self.basis_matrix_[:, :n_basis_modes].copy()#self.original_data @
         else:
-            return self.basis_matrix_[:, :n_basis_modes]
+            return self.basis_matrix_[:, :n_basis_modes]#self.original_data @
 
     def _validate_input(self, n_basis_modes):
         """

pysensors/basis/_identity.py

@@ -6,6 +6,7 @@
 from warnings import warn
 
 from numpy import identity
+import numpy as np
 from sklearn.base import BaseEstimator
 from sklearn.utils import check_array
 
@@ -52,7 +53,8 @@ def fit(self, X):
         -------
         self : instance
         """
-
+        # Store original data
+        self.original_data = X
         # Note that we take a transpose here, so columns correspond to examples
         if self.n_basis_modes is None:
             self.basis_matrix_ = check_array(X).T.copy()
@@ -65,10 +67,10 @@ def fit(self, X):
                     )
                 )
 
-            self.basis_matrix_ = check_array(X)[: self.n_basis_modes, :].T.copy()
+            self.basis_matrix_ = np.eye(X.shape[1])[:,:self.n_basis_modes] #check_array(X)[: self.n_basis_modes, :].T.copy()
 
-            if self.n_basis_modes < X.shape[0]:
-                warn(f"Only the first {self.n_basis_modes} examples were retained.")
+            # if self.n_basis_modes < X.shape[0]:
+            #     warn(f"Only the first {self.n_basis_modes} examples were retained.")
         return self
 
     def matrix_inverse(self, n_basis_modes=None):