Merge pull request #2175 from sechkova/neuralcleanse-nb

Update Neural Cleanse notebook imports
Trusted-AI · Jun 13, 2023 · cc89606 · cc89606
2 parents d68eec0 + 7829d1b
commit cc89606
Showing 1 changed file with 5 additions and 1 deletion.
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()"
    ]
   },
   {