Skip to content

Commit 6af86da

Browse files
authored
update onehotencoder parameters (#253)
1 parent 3fbeac3 commit 6af86da

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

setup.cfg

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ install_requires =
3535
importlib-metadata
3636
pandas>=1.4,<2
3737
torch>=1.10.0,<2.0
38-
scikit-learn>=1.0
38+
scikit-learn>=1.2
3939
nflows>=0.14
4040
numpy>=1.20, <1.24
4141
lifelines>=0.27,!= 0.27.5

src/synthcity/plugins/core/models/tabular_encoder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ class TabularEncoder(TransformerMixin, BaseEstimator):
6060

6161
categorical_encoder: Union[str, type] = "onehot"
6262
continuous_encoder: Union[str, type] = "bayesian_gmm"
63-
cat_encoder_params: dict = dict(handle_unknown="ignore", sparse=False)
63+
cat_encoder_params: dict = dict(handle_unknown="ignore", sparse_output=False)
6464
cont_encoder_params: dict = dict(n_components=10)
6565

6666
@validate_arguments(config=dict(arbitrary_types_allowed=True))

0 commit comments

Comments
 (0)