IBA icon indicating copy to clipboard operation
IBA copied to clipboard

Some changes for the PyTorch IBA API

Open berleon opened this issue 5 years ago • 0 comments

A few remarks to the PyTorch IBA API:

  • [x] IBA.heatmap rename to IBA.analyze. Add flag to switch between saliency_map (in bits, channels summed, scaled to input image) or capacity (in bits, not summed, not scaled)
  • [x] IBA.supress_information rename to IBA.restrict_flow
  • [x] IBA.__init__ add estimator=None parameter
  • [x] IBA.__init__ add feature_mean_std=None parameter which takes a tuple(mean, std)
  • [x] IBA._init rename to IBA._build: make clear that it is different to __init__

berleon avatar Mar 04 '20 13:03 berleon