Skip to content

Commit

Permalink
Use shadow data generation in model names.
Browse files Browse the repository at this point in the history
To detect wether retraining is necessary
  • Loading branch information
Oli-Ross committed Jan 29, 2024
1 parent fef5e01 commit 7005d85
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion mia/shadow_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,8 +23,9 @@ def set_seed(new_seed: int):

def get_shadow_model_name(config: Dict, i: int):
numModels: int = config["shadowModels"]["number"]
method: str = config["shadowDataset"]["method"]
split: float = config["shadowModels"]["split"]
return "shadow_" + tm.get_model_name(config) + f"_split_{split}_{i+1}_of_{numModels}"
return "shadow_" + method + tm.get_model_name(config) + f"_split_{split}_{i+1}_of_{numModels}"


def load_shadow_models_and_datasets(config: Dict) -> Tuple[List[tm.Sequential], List[Tuple[ds.Dataset, ds.Dataset]]]:
Expand Down

0 comments on commit 7005d85

Please sign in to comment.