From 007367fc7474a91aa714bbcea78f8da092f55aaf Mon Sep 17 00:00:00 2001 From: Abhishek Roy Date: Fri, 1 Nov 2024 17:57:49 +0000 Subject: [PATCH] Update deprecated parameter name `sparse` to `sparse_output` for sklearn's OneHotEncoder --- neurocombat_sklearn/neurocombat_sklearn.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/neurocombat_sklearn/neurocombat_sklearn.py b/neurocombat_sklearn/neurocombat_sklearn.py index 5a68e12..b9d586d 100644 --- a/neurocombat_sklearn/neurocombat_sklearn.py +++ b/neurocombat_sklearn/neurocombat_sklearn.py @@ -188,7 +188,7 @@ def _make_design_matrix(self, sites, discrete_covariates, continuous_covariates, # Sites if fitting: - self.site_encoder = OneHotEncoder(sparse=False) + self.site_encoder = OneHotEncoder(sparse_output=False) self.site_encoder.fit(sites) sites_design = self.site_encoder.transform(sites) @@ -202,7 +202,7 @@ def _make_design_matrix(self, sites, discrete_covariates, continuous_covariates, self.discrete_encoders = [] for i in range(n_discrete_covariates): - discrete_encoder = OneHotEncoder(sparse=False) + discrete_encoder = OneHotEncoder(sparse_output=False) discrete_encoder.fit(discrete_covariates[:, i][:, np.newaxis]) self.discrete_encoders.append(discrete_encoder)