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

Merge with main

Browse source
This commit is contained in:
abigailt 2022-08-01 18:12:03 +03:00
parent dc5cc793ee
commit 64038f76f9
2 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -273,9 +273,9 @@ class DatasetWithPredictions(Dataset):
y: Optional[INPUT_DATA_ARRAY_TYPE] = None, features_names: Optional[list] = None, **kwargs):
self.is_pandas = False
self.features_names = features_names
self._pred = self._array2numpy(pred)
self._y = self._array2numpy(y) if y is not None else None
self._x = self._array2numpy(x) if x is not None else None
self._pred = array2numpy(pred)
self._y = array2numpy(y) if y is not None else None
self._x = array2numpy(x) if x is not None else None
if self.is_pandas and x is not None:
if features_names and not np.array_equal(features_names, x.columns):
raise ValueError("The supplied features are not the same as in the data features")