Extend source_set to #source_num samples, allowing more samples to poison
(source_set, source_num)
| 237 | return poison_set, poison_ids |
| 238 | |
| 239 | def extend_source(source_set, source_num): |
| 240 | "Extend source_set to #source_num samples, allowing more samples to poison" |
| 241 | if source_num==0: # if source num is set to 0, means doesn't need extend source set |
| 242 | return source_set |
| 243 | else: |
| 244 | new_source_set = source_set |
| 245 | while len(new_source_set)<source_num: |
| 246 | new_source_set.extend(source_set) |
| 247 | return new_source_set[:source_num] |
| 248 | |
| 249 | def prepare_dataset(source_num, trainset, testset, y_target, y_source, patch, random_patch): |
| 250 | """ prepare benign datasets and source datasets and patched(poisoned) datasets""" |