search

memgraph / mage

MAGE - Memgraph Advanced Graph Extensions :crystal_ball:
Apache License 2.0
250 stars 25 forks source link

[BUG] tgn.predict_link_score returning values greater than 1 #455

Open gefend opened 6 months ago

gefend commented 6 months ago

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.

image

antejavor commented 6 months ago

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