diff --git a/ppca/_ppca.py b/ppca/_ppca.py index 45e5f2b..0dcc251 100644 --- a/ppca/_ppca.py +++ b/ppca/_ppca.py @@ -108,6 +108,9 @@ def fit(self, data, d=None, tol=1e-4, min_obs=10, verbose=False): self.eig_vals = vals self._calc_var() + # to allow using this as a part of sklearn's pipeline + return self + def transform(self, data=None): if self.C is None: