Add feature ablation to Insights

Author: edward-io

captum/insights/config.py

@@ -4,6 +4,7 @@
 from captum.attr import (
     Deconvolution,
     DeepLift,
+    FeatureAblation,
     GuidedBackprop,
     InputXGradient,
     IntegratedGradients,
@@ -31,6 +32,7 @@ class StrEnumConfig(NamedTuple):
     InputXGradient,
     IntegratedGradients,
     Saliency,
+    FeatureAblation,
 ]
 
 ATTRIBUTION_NAMES_TO_METHODS = {
@@ -43,5 +45,8 @@ class StrEnumConfig(NamedTuple):
     IntegratedGradients.get_name(): {
         "n_steps": NumberConfig(value=25, limit=(2, None)),
         "method": StrEnumConfig(limit=SUPPORTED_METHODS, value="gausslegendre"),
-    }
+    },
+    FeatureAblation.get_name(): {
+        "perturbations_per_eval": NumberConfig(value=1, limit=(1, 100)),
+    },
 }