model_inversion.py [244:251]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    return results


def main(args):
    regression = (args.dataset == "iwpc" or args.dataset == "synth")
    data = dataloading.load_dataset(
        name=args.dataset, split="train", normalize=False,
        num_classes=2, root=args.data_folder, regression=regression)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



private_model_inversion.py [97:104]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    return results


def main(args):
    regression = (args.dataset == "iwpc" or args.dataset == "synth")
    data = dataloading.load_dataset(
        name=args.dataset, split="train", normalize=False,
        num_classes=2, root=args.data_folder, regression=regression)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -