Unverified Commit 22d0df8b authored by Moritz Blum's avatar Moritz Blum Committed by GitHub
Browse files

[Doc] Update link_predict.py docstring (#2960)



corrected docstring, the model uses two R-GCN layers
Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
parent 86ba2335
......@@ -6,9 +6,7 @@ Code: https://github.com/MichSchli/RelationPrediction
Difference compared to MichSchli/RelationPrediction
* Report raw metrics instead of filtered metrics.
* By default, we use uniform edge sampling instead of neighbor-based edge
sampling used in author's code. In practice, we find it achieves similar MRR
probably because the model only uses one GNN layer so messages are propagated
among immediate neighbors. User could specify "--edge-sampler=neighbor" to switch
sampling used in author's code. In practice, we find it achieves similar MRR. User could specify "--edge-sampler=neighbor" to switch
to neighbor-based edge sampling.
"""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment