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

BlackboxClassifier based on predictions to work with DatasetWithPredictions

Browse source
This commit is contained in:
abigailt 2022-07-24 21:05:12 +03:00 • committed by abigailgold
parent 77a6e08c8e
commit fb534f7a0f
3 changed files with 63 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -477,7 +477,7 @@ class Data:
def get_train_samples(self) -> Collection[Any]:
"""
Get train set samples
Get train set samples, or None if no training data provided
:return: training samples
"""
@ -487,7 +487,7 @@ class Data:
def get_train_labels(self) -> Collection[Any]:
"""
Get train set labels
Get train set labels, or None if no training labels provided
:return: training labels
"""
@ -495,6 +495,16 @@ class Data:
return None
return self.train.get_labels()
def get_train_predictions(self) -> Collection[Any]:
"""
Get train set predictions, or None if no training predictions provided
:return: training labels
"""
if self.train is None:
return None
return self.train.get_predictions()
def get_test_samples(self) -> Collection[Any]:
"""
Get test set samples
@ -509,8 +519,18 @@ class Data:
"""
Get test set labels
:return: test labels, or None if no test data provided
:return: test labels, or None if no test labels provided
"""
if self.test is None:
return None
return self.test.get_labels()
def get_test_predictions(self) -> Collection[Any]:
"""
Get test set predictions, or None if no test predictions provided
:return: test labels
"""
if self.test is None:
return None
return self.test.get_predictions()