mirror of
https://github.com/IBM/ai-privacy-toolkit.git
synced 2026-06-17 15:35:13 +02:00
24 lines
439 B
Python
24 lines
439 B
Python
from dataclasses import dataclass
|
|
from typing import Optional
|
|
|
|
import numpy as np
|
|
|
|
DEFAULT_DATASET_NAME = "dataset"
|
|
|
|
|
|
@dataclass
|
|
class DatasetAttackResult:
|
|
pass
|
|
|
|
|
|
@dataclass
|
|
class DatasetAttackScore:
|
|
dataset_name: str
|
|
risk_score: float
|
|
result: Optional[DatasetAttackResult]
|
|
|
|
|
|
@dataclass
|
|
class DatasetAttackResultMembership(DatasetAttackResult):
|
|
member_probabilities: np.ndarray
|
|
non_member_probabilities: np.ndarray
|