make JSMA tutorial test faster
@@ -11,12 +11,14 @@ class TestMNISTTutorialJSMA(unittest.TestCase):
'train_end': 10000,
'test_start': 0,
'test_end': 1666,
- 'viz_enabled': False}
+ 'viz_enabled': False,
+ 'source_samples': 1,
+ 'nb_epochs': 2}
report = mnist_tutorial_jsma.mnist_tutorial_jsma(**jsma_tutorial_args)
# Check accuracy values contained in the AccuracyReport object
- self.assertTrue(report.clean_train_clean_eval > 0.85)
- self.assertTrue(report.clean_train_adv_eval < 0.15)
+ self.assertTrue(report.clean_train_clean_eval > 0.75)
+ self.assertTrue(report.clean_train_adv_eval < 0.05)
# There is no adversarial training in the JSMA tutorial
self.assertTrue(report.adv_train_clean_eval == 0.)
