IBA
IBA copied to clipboard
Some changes for the PyTorch IBA API
A few remarks to the PyTorch IBA API:
- [x]
IBA.heatmaprename toIBA.analyze. Add flag to switch betweensaliency_map(in bits, channels summed, scaled to input image) orcapacity(in bits, not summed, not scaled) - [x]
IBA.supress_informationrename toIBA.restrict_flow - [x]
IBA.__init__addestimator=Noneparameter - [x]
IBA.__init__addfeature_mean_std=Noneparameter which takes atuple(mean, std) - [x]
IBA._initrename toIBA._build: make clear that it is different to__init__