mirror of
https://github.com/IBM/ai-privacy-toolkit.git
synced 2026-07-14 16:32:12 +02:00
26 lines
487 B
Python
26 lines
487 B
Python
from dataclasses import dataclass, field
|
|
|
|
import numpy as np
|
|
|
|
DEFAULT_DATASET_NAME = "dataset"
|
|
|
|
|
|
@dataclass
|
|
class DatasetAttackScore:
|
|
dataset_name: str
|
|
|
|
|
|
@dataclass
|
|
class DatasetAttackResult:
|
|
pass
|
|
|
|
|
|
@dataclass(repr=False)
|
|
class DatasetAttackScoreWithResult(DatasetAttackScore):
|
|
result: DatasetAttackResult = field(repr=False)
|
|
|
|
|
|
@dataclass
|
|
class DatasetAttackResultMembership(DatasetAttackResult):
|
|
member_probabilities: np.ndarray
|
|
non_member_probabilities: np.ndarray
|