Merge pull request #2175 from sechkova/neuralcleanse-nb

beat-buesser · web-flow · commit cc8960629669 · 2023-06-14T00:05:17.000+01:00
Update Neural Cleanse notebook imports
diff --git a/notebooks/poisoning_defense_neural_cleanse.ipynb b/notebooks/poisoning_defense_neural_cleanse.ipynb
@@ -55,7 +55,11 @@
     "from art.utils import load_mnist, preprocess\n",
     "from art.defences.detector.poison import ActivationDefence\n",
     "from art.defences.transformer.poisoning import NeuralCleanse\n",
-    "from art.estimators.certification.neural_cleanse import KerasNeuralCleanse\n"
+    "from art.estimators.poison_mitigation import KerasNeuralCleanse\n",
+    "\n",
+    "import tensorflow as tf\n",
+    "if tf.executing_eagerly():\n",
+    "    tf.compat.v1.disable_eager_execution()"
    ]
   },
   {

Original file line number	Diff line number	Diff line change
`@@ -55,7 +55,11 @@`
`55`	`55`	`"from art.utils import load_mnist, preprocess\n",`
`56`	`56`	`"from art.defences.detector.poison import ActivationDefence\n",`
`57`	`57`	`"from art.defences.transformer.poisoning import NeuralCleanse\n",`
`58`		`- "from art.estimators.certification.neural_cleanse import KerasNeuralCleanse\n"`
	`58`	`+ "from art.estimators.poison_mitigation import KerasNeuralCleanse\n",`
	`59`	`+ "\n",`
	`60`	`+ "import tensorflow as tf\n",`
	`61`	`+ "if tf.executing_eagerly():\n",`
	`62`	`+ " tf.compat.v1.disable_eager_execution()"`
`59`	`63`	`]`
`60`	`64`	`},`
`61`	`65`	`{`