mirror of
https://github.com/IBM/ai-privacy-toolkit.git
synced 2026-04-26 13:26:21 +02:00
Increase version to 0.2.0 (#74)
* Remove tensorflow dependency if not using keras model * Remove xgboost dependency if not using xgboost model * Documentation updates Signed-off-by: abigailt <abigailt@il.ibm.com>
This commit is contained in:
parent
782edabd58
commit
8a9ef80146
25 changed files with 306 additions and 152 deletions
|
|
@ -8,11 +8,21 @@ DEFAULT_DATASET_NAME = "dataset"
|
|||
|
||||
@dataclass
|
||||
class DatasetAttackResult:
|
||||
"""
|
||||
Basic class for storing privacy risk assessment results.
|
||||
"""
|
||||
pass
|
||||
|
||||
|
||||
@dataclass
|
||||
class DatasetAttackScore:
|
||||
"""
|
||||
Basic class for storing privacy risk assessment scores.
|
||||
|
||||
:param dataset_name: The name of the dataset that was assessed.
|
||||
:param risk_score: The privacy risk score.
|
||||
:param result: An optional list of more detailed results.
|
||||
"""
|
||||
dataset_name: str
|
||||
risk_score: float
|
||||
result: Optional[DatasetAttackResult]
|
||||
|
|
@ -20,5 +30,11 @@ class DatasetAttackScore:
|
|||
|
||||
@dataclass
|
||||
class DatasetAttackResultMembership(DatasetAttackResult):
|
||||
"""
|
||||
Class for storing membership attack results.
|
||||
|
||||
:param member_probabilities: The attack probabilities for member samples.
|
||||
:param non_member_probabilities: The attack probabilities for non-member samples.
|
||||
"""
|
||||
member_probabilities: np.ndarray
|
||||
non_member_probabilities: np.ndarray
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue