Skip to content

Commit

Permalink
rename hp to ht
Browse files Browse the repository at this point in the history
  • Loading branch information
pvk-developer committed Sep 6, 2022
1 parent 4dfad77 commit 24e4314
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions sdv/metadata/table.py
Original file line number Diff line number Diff line change
Expand Up @@ -515,17 +515,17 @@ def _fit_hyper_transformer(self, data, extra_columns):
else:
dtypes[column] = dtype_kind

hp_config = self._get_hypertransformer_config(dtypes)
ht_config = self._get_hypertransformer_config(dtypes)
for column in numerical_extras:
dtypes[column] = 'numerical'
hp_config['sdtypes'][column] = 'numerical'
hp_config['transformers'][column] = rdt.transformers.FloatFormatter(
ht_config['sdtypes'][column] = 'numerical'
ht_config['transformers'][column] = rdt.transformers.FloatFormatter(
missing_value_replacement='mean',
model_missing_values=True,
)

self._hyper_transformer = rdt.HyperTransformer()
self._hyper_transformer.set_config(hp_config)
self._hyper_transformer.set_config(ht_config)
fit_columns = list(dtypes)
if not data[fit_columns].empty:
self._hyper_transformer.fit(data[fit_columns])
Expand Down

0 comments on commit 24e4314

Please sign in to comment.