Location of TSR algorithm in the code

[itsmemala]

Hello, I would like to apply the TSR method to my model but I'm not able to find the exact location of the algorithm in the scripts. Could you point me to the lines that implement the algorithm please? Is it the function - GivenAttGetRescaledSaliency()?
If I understand right, in interpret.py, line 138 masks all features for a time step (per TSR algorithm) and therefore should be
'mask[i,:] = 0' and not 'mask[i,:] = i' ?