(self)
| 671 | train, valid, test |
| 672 | """ |
| 673 | def __init__(self): |
| 674 | super(WN18RR, self).__init__( |
| 675 | "wn18rr", |
| 676 | urls={ |
| 677 | "train": "https://github.com/DeepGraphLearning/KnowledgeGraphEmbedding/raw/master/data/wn18rr/train.txt", |
| 678 | "valid": "https://github.com/DeepGraphLearning/KnowledgeGraphEmbedding/raw/master/data/wn18rr/valid.txt", |
| 679 | "test": "https://github.com/DeepGraphLearning/KnowledgeGraphEmbedding/raw/master/data/wn18rr/test.txt" |
| 680 | } |
| 681 | ) |
| 682 | |
| 683 | |
| 684 | class Wikidata5m(Dataset): |