22 lines (16 loc) · 545 Bytes

NYUMLSecLab4

Author: Ankit Rajvanshi (ar7996)

Need to Change paths of following files before running jupyter notebook:

clean_test_data_path
clean_valid_data_path
poisioned_test_data_path
bd_model_path
bd_weights_path

You may remove cell with unzip command depending upon how your data is stored

Requirements:

tensorflow
numpy
h5py
matplotlib

Evaluation:

Repaired Models are present in RepairedModels directory with name <BadNets_prime_{x}percent.h5> where x represents threshold (2, 4 10)