captum
captum copied to clipboard
refactor feature ablation
- refactor to feature ablation to make the logic about multi-output attribution more readable
- removed some (outdated) misleading comments
- add more comments to explain the steps and tensor shapes