[BUG] tgn.predict_link_score returning values greater than 1

[gefend]

I tried runing the notebook amazon_network_analysis.ipynb. The predicted scores returned by the function tgn.predict_link_score were greater than 1. In the offical tgn implemantation, they used a sigmoid after the mlp layer.

[antejavor]

Hi @gefend, thanks for reporting this 🚀 , we will look into this.