apunkt/ai-privacy-toolkit
1
0
Fork 0
mirror of https://github.com/IBM/ai-privacy-toolkit.git synced 2026-07-02 16:01:00 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

formatting

Browse source 
Signed-off-by: abigailt <abigailt@il.ibm.com>
This commit is contained in:
abigailt 2023-08-10 13:20:28 +03:00
parent f85fc87bdd
commit 69e45d99e5
4 changed files with 17 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

2
apt/minimization/minimizer.py
View file

@ -878,7 +878,6 @@ class GeneralizeToRepresentative(BaseEstimator, MetaEstimatorMixin, TransformerM
rows[feature] = generalizations['range_representatives'][feature][r_index]
return original_data_generalized
def _generalize_from_tree(self, original_data, prepared_data, level_nodes, cells, cells_by_id):
mapping_to_cells = self._map_to_cells(prepared_data, level_nodes, cells_by_id)
all_indexes = []
@ -1199,6 +1198,7 @@ class GeneralizeToRepresentative(BaseEstimator, MetaEstimatorMixin, TransformerM
else:
ranges[feature] = list(set(ranges[feature]))
ranges[feature].sort()
prev_value = 0
for index, value in enumerate(ranges[feature]):
if index == 0:
# for first range, use min value