Only impute eigenvecs/eigenvals for eigsh when necessary

biocore · Mar 5, 2018 · 6d59299 · 6d59299
1 parent 8e068a7
commit 6d59299
Showing 1 changed file with 13 additions and 10 deletions.
diff --git a/mdsa/algorithms/eigsh.py b/mdsa/algorithms/eigsh.py
@@ -60,15 +60,18 @@ def _impute_eigendecomposition_results(eigenvectors, eigenvalues,
         eigenvalues = np.array(eigenvalues)
         eigenvectors = np.array(eigenvectors)
 
-        # impute missing eigenvectors, composed of NaN entries
-        nan_eigenvecs = np.array([[np.nan] * eigenvectors.shape[1]] *
-                                 (expected_num_results -
-                                  len(eigenvectors)))
-        eigenvectors = np.concatenate((eigenvectors, nan_eigenvecs),
-                                      axis=0)
+        if len(eigenvectors) != expected_num_results:
+            # impute missing eigenvectors, composed of NaN entries
+            nan_eigenvecs = np.array([[np.nan] * eigenvectors.shape[1]] *
+                                     (expected_num_results -
+                                      len(eigenvectors)))
+            eigenvectors = np.concatenate((eigenvectors, nan_eigenvecs),
+                                          axis=0)
+
+        if len(eigenvalues) != expected_num_results:
+            # impute missing eigenvalues as NaNs
+            nan_eigenvals = [[np.nan] * (expected_num_results -
+                                         len(eigenvalues))]
+            eigenvalues = np.append(eigenvalues, nan_eigenvals)
 
-        # impute missing eigenvalues as NaNs
-        nan_eigenvals = [[np.nan] * (expected_num_results -
-                                     len(eigenvalues))]
-        eigenvalues = np.append(eigenvalues, nan_eigenvals)
         return eigenvectors, eigenvalues