Merge pull request #51 from keesmulder/patch-1

Also re-order the structure matrix
EducationalTestingService · Jun 2, 2020 · 9d712e4 · 9d712e4
2 parents d3df9c6 + 6c07ffb
commit 9d712e4
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/factor_analyzer/factor_analyzer.py b/factor_analyzer/factor_analyzer.py
@@ -655,8 +655,12 @@ def fit(self, X, y=None):
 
         # resort the factors according to their variance
         variance = self._get_factor_variance(loadings)[0]
-        loadings = loadings[:, list(reversed(np.argsort(variance)))].copy()
-
+        new_order = list(reversed(np.argsort(variance)))
+        loadings = loadings[:, new_order].copy()
+
+        if structure is not None: 
+            structure = structure[:, new_order].copy()
+
         self.phi_ = phi
         self.structure_ = structure