Skip to content

Some changes for the PyTorch IBA API #9

@berleon

Description

@berleon

A few remarks to the PyTorch IBA API:

  • 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)
  • IBA.supress_information rename to IBA.restrict_flow
  • IBA.__init__ add estimator=None parameter
  • IBA.__init__ add feature_mean_std=None parameter which takes a tuple(mean, std)
  • IBA._init rename to IBA._build: make clear that it is different to __init__

Metadata

Metadata

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions