add compute_loss example in link prediction user guide (#2487)

4507bebc · Quan (Andy) Gan · GitHub · 0c2a2ea1 · 4507bebc
Unverified Commit 4507bebc authored Jan 06, 2021 by Quan (Andy) Gan Committed by GitHub Jan 06, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

docs/source/guide/minibatch-link.rst docs/source/guide/minibatch-link.rst +5 -0

No files found.
--- a/docs/source/guide/minibatch-link.rst
+++ b/docs/source/guide/minibatch-link.rst
@@ -137,6 +137,11 @@ above.

 .. code:: python

+    def compute_loss(pos_score, neg_score):
+        # an example hinge loss
+        n = pos_score.shape[0]
+        return (neg_score.view(n, -1) - pos_score.view(n, -1) + 1).clamp(min=0).mean()
+
    model = Model(in_features, hidden_features, out_features)
    model = model.cuda()
    opt = torch.optim.Adam(model.parameters())