n_features_in_ consistency for FactorAnalysis

ogrisel · ogrisel · commit 108c7a4d13c7 · 2020-10-07T18:25:44.000+02:00
diff --git a/sklearn/decomposition/_factor_analysis.py b/sklearn/decomposition/_factor_analysis.py
@@ -26,7 +26,7 @@
 
 
 from ..base import BaseEstimator, TransformerMixin
-from ..utils import check_array, check_random_state
+from ..utils import check_random_state
 from ..utils.extmath import fast_logdet, randomized_svd, squared_norm
 from ..utils.validation import check_is_fitted, _deprecate_positional_args
 from ..exceptions import ConvergenceWarning
@@ -279,7 +279,7 @@ def transform(self, X):
         """
         check_is_fitted(self)
 
-        X = check_array(X)
+        X = self._validate_data(X, reset=False)
         Ih = np.eye(len(self.components_))
 
         X_transformed = X - self.mean_
@@ -350,7 +350,7 @@ def score_samples(self, X):
             Log-likelihood of each sample under the current model
         """
         check_is_fitted(self)
-
+        X = self._validate_data(X, copy=self.copy, dtype=np.float64)
         Xr = X - self.mean_
         precision = self.get_precision()
         n_features = X.shape[1]
