advAttack_bug

Author: yyou1996

adversarialRobustness_Component/code/graph_attack/dqn.py

@@ -100,7 +100,7 @@ def eval(self):
             list_at = self.make_actions(t, greedy=True)
             self.env.step(list_at)
             t += 1
-        test_loss = loop_dataset(env.g_list, env.classifier, list(range(len(env.g_list))))
+        test_loss = loop_dataset(env.g_list, env.classifier, list(range(len(env.g_list))), epoch=101)
         print('\033[93m average test: loss %.5f acc %.5f\033[0m' % (test_loss[0], test_loss[1]))
 
         if cmd_args.phase == 'train' and self.best_eval is None or test_loss[1] < self.best_eval:

adversarialRobustness_Component/code/graph_attack/plot_dqn.py

@@ -100,7 +100,7 @@ def eval(self):
             list_at = self.make_actions(t, greedy=True)
             self.env.step(list_at)
             t += 1
-        test_loss = loop_dataset(env.g_list, env.classifier, list(range(len(env.g_list))))
+        test_loss = loop_dataset(env.g_list, env.classifier, list(range(len(env.g_list))), epoch=101)
         print('\033[93m average test: loss %.5f acc %.5f\033[0m' % (test_loss[0], test_loss[1]))
         with open('%s/edge_added.txt' % cmd_args.save_dir, 'w') as f:
             for i in range(len(self.test_g_list)):
@@ -142,4 +142,4 @@ def eval(self):
         # test_loss = loop_dataset(env.g_list, base_classifier, list(range(len(env.g_list))))
         # print('\033[93maverage test: loss %.5f acc %.5f\033[0m' % (test_loss[0], test_loss[1]))
 
-        # print(np.mean(avg_rewards), np.mean(env.rewards))
+        # print(np.mean(avg_rewards), np.mean(env.rewards))

adversarialRobustness_Component/code/graph_classification/er_components.py

@@ -41,7 +41,7 @@
     if cmd_args.ctx == 'gpu':
         classifier = classifier.cuda()
     if cmd_args.phase == 'test':
-        test_loss = loop_dataset(test_glist, classifier, list(range(len(test_glist))))
+        test_loss = loop_dataset(test_glist, classifier, list(range(len(test_glist))), epoch=101)
         print('\033[93maverage test: loss %.5f acc %.5f\033[0m' % (test_loss[0], test_loss[1]))
 
     if cmd_args.phase == 'train':