-
Notifications
You must be signed in to change notification settings - Fork 3k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[Doc] Evaluation Tutorial for Link Prediction #3463
Conversation
To trigger regression tests:
|
@@ -218,8 +218,8 @@ def forward(self, g, h): | |||
|
|||
|
|||
###################################################################### | |||
# Evaluating Performance (Optional) | |||
# --------------------------------- |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I feel this sentence needs improvement. ", where it treats the node embeddings learned by link prediction via training and evaluating a linear classifier on top of the learned node embeddings."
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Rephrased it.
###################################################################### | ||
# First you need to construct a graph for ``dgl.dataloading.EdgeDataLoader`` | ||
# to iterate on, which should have the same nodes as the original | ||
# ``graph``, but the testing node pairs as edges. You also need to |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
A natural question is then what if a user wants to apply a trained GNN to predict edges for a new graph with a different set of nodes?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Extra empty space after "edges."
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Good suggestion. I rephrased it so I'm not emphasizing that the graph should be the same as original.
Is there any problems in link-prediction-eval branch? |
Description
Add a tutorial section for evaluating link prediction models on link prediction task.