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

Working example of anonymization with pytorch multi-output binary model

Browse source 
Signed-off-by: abigailt <abigailt@il.ibm.com>
This commit is contained in:
abigailt 2024-03-12 13:25:31 +02:00
parent 5e19d4ae27
commit 076503b248
2 changed files with 76 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
apt/utils/datasets/datasets.py
View file

@ -233,7 +233,7 @@ class ArrayDataset(Dataset):
raise ValueError("The supplied features are not the same as in the data features")
self.features_names = x.columns.to_list()
if self._y is not None and len(self._x) != len(self._y):
if self._y is not None and self._x.shape[0] != self._y.shape[0]:
raise ValueError("Non equivalent lengths of x and y")
def get_samples(self) -> OUTPUT_DATA_ARRAY_TYPE: