search

somepago / saint

The official PyTorch implementation of recent paper - SAINT: Improved Neural Networks for Tabular Data via Row Attention and Contrastive Pre-Training
Apache License 2.0
402 stars 63 forks source link

Code for interpretability #8

Open Fabien-Couthouis opened 3 years ago

Fabien-Couthouis commented 3 years ago

Hello,

Thanks for this repo! Do you plan to release the code you used to plot intersample attention and self attention as in the paper (section 5.2)? I would like to reproduce the figure 3.

somepago commented 3 years ago

Hi Fabien, I will release the attention plotting code in the next version. I am busy with another project rn, I am targeting the end of November for this. If you need it urgently let me know.

Fabien-Couthouis commented 3 years ago

Thanks for your answer! I understand your time is precious but it would help me a lot if you could just share the code for attention plots.

bdubey commented 2 years ago

@somepago Wold be great help if you can share the code for interpretability. I have reached out to you by email as well.

isamgul commented 2 years ago

Hello Gowthami,

Thanks for this project. It will be of great help to get the attention plotting code for the SAINT implementation.