Skip to content

Commit

Permalink
[Doc] Clarify link prediction tutorial (dmlc#2673)
Browse files Browse the repository at this point in the history
* link prediction clarification

* fix
  • Loading branch information
BarclayII authored Feb 23, 2021
1 parent 7298173 commit da2f690
Showing 1 changed file with 7 additions and 3 deletions.
10 changes: 7 additions & 3 deletions tutorials/blitz/4_link_predict.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,9 +37,7 @@
# either citing or being cited, between two papers exists in a citation
# network.
#
# This tutorial follows a relatively simple practice from
# `SEAL <https://papers.nips.cc/paper/2018/file/53f0d7c537d99b3824f0f99d62ea2428-Paper.pdf>`__.
# It formulates the link prediction problem as a binary classification
# This tutorial formulates the link prediction problem as a binary classification
# problem as follows:
#
# - Treat the edges in the graph as *positive examples*.
Expand All @@ -50,6 +48,12 @@
# - Evaluate the model with any binary classification metric such as Area
# Under Curve (AUC).
#
# .. note::
#
# The practice comes from
# `SEAL <https://papers.nips.cc/paper/2018/file/53f0d7c537d99b3824f0f99d62ea2428-Paper.pdf>`__,
# although the model here does not use their idea of node labeling.
#
# In some domains such as large-scale recommender systems or information
# retrieval, you may favor metrics that emphasize good performance of
# top-K predictions. In these cases you may want to consider other metrics
Expand Down

0 comments on commit da2f690

Please sign in to comment.