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

Support 1-hot encoded features in anonymization + fixes related to encoding in minimization (#86)

Browse source 
* Support 1-hot encoded features in anonymization (#72)
* Fix anonymization adult notebook + new notebook to demonstrate anonymization on 1-hot encoded data

* Minimizer: No default encoder, if none provided data is supplied to the model as is. Fix data type of representative values. Fix and add more tests.

Signed-off-by: abigailt <abigailt@il.ibm.com>
This commit is contained in:
abigailgold 2023-10-19 11:48:15 +03:00 • committed by GitHub
parent 26addd192f
commit 5dce961092
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 670 additions and 255 deletions
Download patch file Download diff file Expand all files Collapse all files

6
apt/utils/models/pytorch_model.py
View file

@ -368,7 +368,7 @@ class PyTorchClassifier(PyTorchModel):
if validation_data is None:
self._art_model.fit(
x=train_data.get_samples(),
y=train_data.get_labels().reshape(-1, 1),
y=train_data.get_labels(),
batch_size=batch_size,
nb_epochs=nb_epochs,
save_checkpoints=save_checkpoints,
@ -379,9 +379,9 @@ class PyTorchClassifier(PyTorchModel):
else:
self._art_model.fit(
x=train_data.get_samples(),
y=train_data.get_labels().reshape(-1, 1),
y=train_data.get_labels(),
x_validation=validation_data.get_samples(),
y_validation=validation_data.get_labels().reshape(-1, 1),
y_validation=validation_data.get_labels(),
batch_size=batch_size,
nb_epochs=nb_epochs,
save_checkpoints=save_checkpoints,