Heatmap Visualization

[RaoUmer]

Thanks for providing the greate toolbox.

Attention / Heatmaps visualization is also challenging task for interpretibility of model's results. Would be greate, if it supports into this library in future versions.

Thanks.

[guillaumejaume]

We will push a tutorial for MIL training from scratch with an attention heatmap. For now, please refer to:

• https://github.com/mahmoodlab/TRIDENT/blob/main/tutorials/3-Building-Attention-Heatmaps-for-Your-Model.ipynb