apunkt/ai-privacy-toolkit
1
0
Fork 0
mirror of https://github.com/IBM/ai-privacy-toolkit.git synced 2026-06-08 15:05:13 +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-05-31 14:09:41 +03:00
parent 30725b33b7
commit 7403a0e93a
2 changed files with 11 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

2
apt/minimization/minimizer.py
View file

@ -1107,4 +1107,4 @@ class GeneralizeToRepresentative(BaseEstimator, MetaEstimatorMixin, TransformerM
to_remove.append(feature)
for feature in to_remove:
del generalizations['categories'][feature]
del generalizations['categories'][feature]

20
tests/test_minimizer.py
View file

@ -213,11 +213,11 @@ def test_minimizer_ncp():
gen2.transform(dataset=ad1)
ncp6 = gen2.ncp
assert(ncp1 <= ncp3)
assert(ncp2 != ncp3)
assert(ncp3 != ncp4)
assert(ncp4 != ncp5)
assert(ncp6 == ncp4)
assert (ncp1 <= ncp3)
assert (ncp2 != ncp3)
assert (ncp3 != ncp4)
assert (ncp4 != ncp5)
assert (ncp6 == ncp4)
def test_minimizer_ncp_categorical():
@ -289,11 +289,11 @@ def test_minimizer_ncp_categorical():
gen2.transform(dataset=ad1)
ncp6 = gen2.ncp
assert(ncp1 <= ncp3)
assert(ncp2 != ncp3)
assert(ncp3 != ncp4)
assert(ncp4 != ncp5)
assert(ncp6 == ncp4)
assert (ncp1 <= ncp3)
assert (ncp2 != ncp3)
assert (ncp3 != ncp4)
assert (ncp4 != ncp5)
assert (ncp6 == ncp4)
def test_minimizer_fit_not_transform():