mirror of
https://github.com/IBM/ai-privacy-toolkit.git
synced 2026-07-02 16:01:00 +02:00
26 lines
486 B
Python
26 lines
486 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 DatasetAttackResultPerRecord(DatasetAttackResult):
|
|
positive_probabilities: np.ndarray
|
|
negative_probabilities: np.ndarray
|