Update kpca.py

fix to cover selected components
yzhao062 · Nov 14, 2023 · e78400f · e78400f
1 parent 819c8fc
commit e78400f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pyod/models/kpca.py b/pyod/models/kpca.py
@@ -331,6 +331,7 @@ def fit(self, X, y=None):
             random_state=self.random_state,
         )
         x_transformed = self.kpca.fit_transform(X)
+        x_transformed = x_transformed[:, : self.n_selected_components_]
 
         centerer = self.kpca.get_centerer
         kernel = self.kpca.get_kernel
@@ -375,6 +376,8 @@ def decision_function(self, X):
         gram_matrix = kernel(X, self.kpca.X_fit_)
 
         x_transformed = self.kpca.transform(X)
+        x_transformed = x_transformed[:, : self.n_selected_components_]
+
         potential = []
         for i in range(X.shape[0]):
             sample = X[i, :].reshape(1, -1)