generalize_using_transform=False supported

Signed-off-by: abigailt <abigailt@il.ibm.com>
This commit is contained in:
abigailt 2023-05-18 10:32:54 +03:00
parent 51340fa554
commit 4a252e65fe
2 changed files with 32 additions and 4 deletions

View file

@ -93,8 +93,7 @@ class GeneralizeToRepresentative(BaseEstimator, MetaEstimatorMixin, TransformerM
self.train_only_features_to_minimize = train_only_features_to_minimize
self.is_regression = is_regression
self.encoder = encoder
# self.generalize_using_transform = generalize_using_transform
self.generalize_using_transform = False
self.generalize_using_transform = generalize_using_transform
self._ncp = 0.0
self._feature_data = {}
self._categorical_values = {}