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)