From 0765009cad31a140ff53d224e5fcee65aa1d93d6 Mon Sep 17 00:00:00 2001 From: hnurxn <84432936+hnurxn@users.noreply.github.com> Date: Tue, 28 Jun 2022 12:31:16 +0800 Subject: [PATCH] Update integrated_gradients.py completion the missing formulas add x-x' --- tf_explain/core/integrated_gradients.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tf_explain/core/integrated_gradients.py b/tf_explain/core/integrated_gradients.py index 9308b30..ade4408 100644 --- a/tf_explain/core/integrated_gradients.py +++ b/tf_explain/core/integrated_gradients.py @@ -42,7 +42,7 @@ def explain(self, validation_data, model, class_index, n_steps=10): ) grayscale_integrated_gradients = transform_to_normalized_grayscale( - tf.abs(integrated_gradients) + tf.abs(images*integrated_gradients) ).numpy() grid = grid_display(grayscale_integrated_gradients)