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

add minimization notebook (#22)

Browse source 
* add german credit notebook to showcase new features (minimize only some features and categorical features)

* add notebook to show minimization data on a regression problem
This commit is contained in:
olasaadi 2022-02-23 14:57:12 +02:00 • committed by GitHub
parent 9de078f937
commit 752871dd0c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 648 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/test_minimizer.py
View file

@ -510,7 +510,7 @@ def test_regression():
transformed = gen.transform(x_train)
print('Base model accuracy (R2 score): ', model.score(x_test, y_test))
model.fit(transformed, y_train)
print('Base model accuracy (R2 score) after anonymization: ', model.score(x_test, y_test))
print('Base model accuracy (R2 score) after minimization: ', model.score(x_test, y_test))
gener = gen.generalizations_
expexted_generalizations = {'ranges': {
'age': [-0.07816532626748085, -0.07090024650096893, -0.05637009255588055, -0.05092128552496433,