captum/attr/__init__.py

#!/usr/bin/env python3
from captum.attr._core.deep_lift import DeepLift, DeepLiftShap  # noqa
from captum.attr._core.feature_ablation import FeatureAblation  # noqa
from captum.attr._core.feature_permutation import FeaturePermutation  # noqa
from captum.attr._core.gradient_shap import GradientShap  # noqa
from captum.attr._core.guided_backprop_deconvnet import (  # noqa
    Deconvolution,
    GuidedBackprop,
)
from captum.attr._core.guided_grad_cam import GuidedGradCam  # noqa
from captum.attr._core.input_x_gradient import InputXGradient  # noqa
from captum.attr._core.integrated_gradients import IntegratedGradients  # noqa
from captum.attr._core.kernel_shap import KernelShap  # noqa
from captum.attr._core.layer.grad_cam import LayerGradCam  # noqa
from captum.attr._core.layer.internal_influence import InternalInfluence  # noqa
from captum.attr._core.layer.layer_activation import LayerActivation  # noqa
from captum.attr._core.layer.layer_conductance import LayerConductance  # noqa
from captum.attr._core.layer.layer_deep_lift import (  # noqa
    LayerDeepLift,
    LayerDeepLiftShap,
)
from captum.attr._core.layer.layer_feature_ablation import LayerFeatureAblation  # noqa
from captum.attr._core.layer.layer_gradient_shap import LayerGradientShap  # noqa
from captum.attr._core.layer.layer_gradient_x_activation import (  # noqa
    LayerGradientXActivation,
)
from captum.attr._core.layer.layer_integrated_gradients import (  # noqa
    LayerIntegratedGradients,
)
from captum.attr._core.layer.layer_lrp import LayerLRP  # noqa
from captum.attr._core.lime import Lime, LimeBase  # noqa
from captum.attr._core.lrp import LRP  # noqa
from captum.attr._core.neuron.neuron_conductance import NeuronConductance  # noqa
from captum.attr._core.neuron.neuron_deep_lift import (  # noqa
    NeuronDeepLift,
    NeuronDeepLiftShap,
)
from captum.attr._core.neuron.neuron_feature_ablation import (  # noqa
    NeuronFeatureAblation,
)
from captum.attr._core.neuron.neuron_gradient import NeuronGradient  # noqa
from captum.attr._core.neuron.neuron_gradient_shap import NeuronGradientShap  # noqa
from captum.attr._core.neuron.neuron_guided_backprop_deconvnet import (  # noqa
    NeuronDeconvolution,
    NeuronGuidedBackprop,
)
from captum.attr._core.neuron.neuron_integrated_gradients import (  # noqa
    NeuronIntegratedGradients,
)
from captum.attr._core.noise_tunnel import NoiseTunnel  # noqa
from captum.attr._core.occlusion import Occlusion  # noqa
from captum.attr._core.saliency import Saliency  # noqa
from captum.attr._core.shapley_value import ShapleyValues, ShapleyValueSampling  # noqa
from captum.attr._models.base import (  # noqa
    configure_interpretable_embedding_layer,
    InterpretableEmbeddingBase,
    remove_interpretable_embedding_layer,
    TokenReferenceBase,
)
from captum.attr._utils import visualization  # noqa
from captum.attr._utils.attribution import (  # noqa  # noqa  # noqa  # noqa  # noqa
    Attribution,
    GradientAttribution,
    LayerAttribution,
    NeuronAttribution,
    PerturbationAttribution,
)
from captum.attr._utils.class_summarizer import ClassSummarizer
from captum.attr._utils.stat import (
    CommonStats,
    Count,
    Max,
    Mean,
    Min,
    MSE,
    StdDev,
    Sum,
    Var,
)
from captum.attr._utils.summarizer import Summarizer

__all__ = [
    "Attribution",
    "GradientAttribution",
    "PerturbationAttribution",
    "NeuronAttribution",
    "LayerAttribution",
    "IntegratedGradients",
    "DeepLift",
    "DeepLiftShap",
    "InputXGradient",
    "Saliency",
    "GuidedBackprop",
    "Deconvolution",
    "GuidedGradCam",
    "FeatureAblation",
    "FeaturePermutation",
    "Occlusion",
    "ShapleyValueSampling",
    "ShapleyValues",
    "LimeBase",
    "Lime",
    "LRP",
    "KernelShap",
    "LayerConductance",
    "LayerGradientXActivation",
    "LayerActivation",
    "LayerFeatureAblation",
    "InternalInfluence",
    "LayerGradCam",
    "LayerDeepLift",
    "LayerDeepLiftShap",
    "LayerGradientShap",
    "LayerIntegratedGradients",
    "LayerLRP",
    "NeuronConductance",
    "NeuronFeatureAblation",
    "NeuronGradient",
    "NeuronIntegratedGradients",
    "NeuronDeepLift",
    "NeuronDeepLiftShap",
    "NeuronGradientShap",
    "NeuronDeconvolution",
    "NeuronGuidedBackprop",
    "NoiseTunnel",
    "GradientShap",
    "InterpretableEmbeddingBase",
    "TokenReferenceBase",
    "visualization",
    "configure_interpretable_embedding_layer",
    "remove_interpretable_embedding_layer",
    "Summarizer",
    "CommonStats",
    "ClassSummarizer",
    "Mean",
    "StdDev",
    "MSE",
    "Var",
    "Min",
    "Max",
    "Sum",
    "Count",
]