fix knockoff

Author: lionelkusch

src/hidimstat/knockoffs.py

@@ -7,10 +7,7 @@
 from sklearn.utils import check_random_state
 from sklearn.utils.validation import check_memory
 
-from hidimstat.gaussian_knockoff import (
-    gaussian_knockoff_generation,
-    repeat_gaussian_knockoff_generation,
-)
+from hidimstat.statistical_tools.gaussian_distribution import GaussianDistribution
 from hidimstat.statistical_tools.multiple_testing import fdr_threshold
 from hidimstat.statistical_tools.aggregation import quantile_aggregation
 
@@ -196,32 +193,12 @@ def model_x_knockoff(
         raise TypeError("Wrong type for random_state")
     seed_list = rng.randint(1, np.iinfo(np.int32).max, n_bootstraps)
 
-    if centered:
-        X = StandardScaler().fit_transform(X)
-
-    # estimation of X distribution
-    # original implementation:
-    # https://github.com/msesia/knockoff-filter/blob/master/R/knockoff/R/create_second_order.R
-    mu = X.mean(axis=0)
-    sigma = cov_estimator.fit(X).covariance_
-
     # Create knockoff variables
-    X_tilde, mu_tilde, sigma_tilde_decompose = memory.cache(
-        gaussian_knockoff_generation
-    )(X, mu, sigma, seed=seed_list[0], tol=tol_gauss)
-
-    if n_bootstraps == 1:
-        X_tildes = [X_tilde]
-    else:
-        X_tildes = parallel(
-            delayed(repeat_gaussian_knockoff_generation)(
-                mu_tilde,
-                sigma_tilde_decompose,
-                seed=seed,
-            )
-            for seed in seed_list[1:]
-        )
-        X_tildes.insert(0, X_tilde)
+    conditionnal_sampler = GaussianDistribution(
+        cov_estimator, random_state=seed_list[0], centered=centered, tol=tol_gauss
+    )
+    conditionnal_sampler.fit(X)
+    X_tildes = [conditionnal_sampler.sample() for i in range(n_bootstraps)]
 
     results = parallel(
         delayed(memory.cache(_stat_coefficient_diff))(