dmlc / dmlc/dgl

Typo in train test split of Link Prediction tutorial

Open
#7,839 1 comment 0 reactions 0 assignees View on GitHub
stale-issue
Dominant language
Python
Stars
14.3k
Forks
3.1k
PR merge metrics
No merged PRs in 30d

Description

## 📚 Documentation

Issue with link prediction tutorial

Hi, in the link prediction tutorial, it seems like there's a typo where you flipped the train and test sets.
for instance in "[Preparing train and test set](https://docs.dgl.ai/en/1.1.x/tutorials/blitz/4_link_predict.html#prepare-training-and-testing-sets)"
```
test_pos_u, test_pos_v = u[eids[:test_size]], v[eids[:test_size]]
train_pos_u, train_pos_v = u[eids[test_size:]], v[eids[test_size:]]
...
train_g = dgl.remove_edges(g, eids[:test_size])
```

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.