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

Wrappers no train (#40)

Browse source 
1) Handle train None in Data
2) Update BB Classifier to handle None either for train or test (x or y)
This commit is contained in:
Shlomit Shachor 2022-06-26 14:43:22 +03:00 • committed by GitHub
parent dfa684da6b
commit 1c4b963add
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 103 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -400,6 +400,8 @@ class Data:
:return: training samples
"""
if self.train is None:
return None
return self.train.get_samples()
def get_train_labels(self) -> Collection[Any]:
@ -408,6 +410,8 @@ class Data:
:return: training labels
"""
if self.train is None:
return None
return self.train.get_labels()
def get_test_samples(self) -> Collection[Any]: