AttentiveFP - attention weight visualization


in the AttentiveFP paper, there’s a part where they visualize the attention weight on the molecular graphs. I was wondering if this has been also implemented in the dgllife framework?

Have you checked the visualization section here?

Fantastic, thank you so much! May I suggest adding a reference in the docs?

Do you mean a reference in the doc of the model?

I meant here somewhere here:

I’ve updated the documentation homepage. See if that looks good to you.

1 Like

Looks great, thank you!