from pgd_attack import LinfPGDAttack os.environ["CUDA_VISIBLE_DEVICES"] = "0" # Global constants with open('config.json') as config_file: config = json.load(config_file) num_eval_examples = config['num_eval_examples'] eval_batch_size = config['eval_batch_size'] data_path = config['data_path'] model_dir = config['model_dir'] # Set upd the data, hyperparameters, and the model flower = flower_input.FlowerData(data_path) model = Model(mode='eval') global_step = tf.contrib.framework.get_or_create_global_step() # Setting up the Tensorboard and checkpoint outputs if not os.path.exists(model_dir): os.makedirs(model_dir) eval_dir = os.path.join(model_dir, 'eval') if not os.path.exists(eval_dir): os.makedirs(eval_dir) saver = tf.train.Saver() summary_writer = tf.summary.FileWriter(eval_dir)
def dataReader(): flower = flower_input.FlowerData('data') X = flower.eval_data.xs y = flower.eval_data.ys return X, y.astype(int)