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

Return None where no predictions are available

Browse source
This commit is contained in:
abigailt 2022-07-12 09:45:15 +03:00 • committed by abigailgold
parent 00f9c16863
commit bc28f7f26a
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -258,7 +258,7 @@ class ArrayDataset(Dataset):
:return: predictions as numpy array
"""
raise NotImplementedError
return None
class DatasetWithPredictions(Dataset):
@ -333,7 +333,7 @@ class PytorchData(Dataset):
if self.is_pandas:
self.features_names = x.columns
if y is not None and len(self._x) != len(self._y):
if self._y is not None and len(self._x) != len(self._y):
raise ValueError('Non equivalent lengths of x and y')
if self._y is not None:
@ -363,7 +363,7 @@ class PytorchData(Dataset):
:return: predictions as numpy array
"""
raise NotImplementedError
return None
def get_sample_item(self, idx: int) -> Tensor:
"""