apunkt/ai-privacy-toolkit
1
0
Fork 0
mirror of https://github.com/IBM/ai-privacy-toolkit.git synced 2026-06-26 15:49:37 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix issue with computed ranges for one-hot encoded features

Browse source 
Signed-off-by: abigailt <abigailt@il.ibm.com>
This commit is contained in:
abigailt 2024-01-17 15:55:06 +02:00
parent d8de062d43
commit 4282cf3208
2 changed files with 46 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
apt/minimization/minimizer.py
View file

@ -576,7 +576,7 @@ class GeneralizeToRepresentative(BaseEstimator, MetaEstimatorMixin, TransformerM
if feature not in feature_data.keys():
fd = {}
values = list(x.loc[:, feature])
if feature not in self.categorical_features:
if feature not in self.categorical_features and feature not in self.all_one_hot_features:
fd['min'] = min(values)
fd['max'] = max(values)
fd['range'] = max(values) - min(values)