Skip to content

Captum v0.3.1 Release
Compare
Choose a tag to compare
@vivekmig vivekmig released this 23 Jan 18:11
v0.3.1
ea1461b
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

Captum v0.3.1 includes some improvements and minor fixes beyond the functionalities added in Captum v0.3.0.

Improvements

Captum v0.3.1 has added improvements to existing attribution methods including:

  • LayerIntegratedGradients now supports computing attributions for multiple layers simultaneously. (PR #532).
  • NoiseTunnel now supports an internal batch size to split noised inputs into batches and appropriately aggregate results (PR #555).
  • visualize_text now has an option return_html to export the visualization as HTML code (PR #548).
  • A utility wrapper was added to allow computing attributions for intermediate layers and inputs simultaneously (PR #534).

Captum Insights

  • Attributions for multiple models can be compared in Captum Insights (PR #551).
  • Various improvements to reduce package size of Captum Insights (PR #556 and #562).

image

Bug Fixes

  • Some parameter names were renamed in NoiseTunnel, Kernel Shap, and Lime to avoid conflicting names when combining Noise Tunnel or metrics with attribution methods. Deprecated arguments now raise warnings and will be removed in 0.4.0 (PR #558).
  • Feature Ablation now supports cases where the output may be on a different device than the input, which may occur in model-parallel setups (#528).
  • Lime (and KernelShap) were fixed to appropriately handle int or long input types (#570).
Assets 2
Loading