WN18 knowledge graph dataset. Splits: train, valid, test
| 646 | |
| 647 | |
| 648 | class WN18(Dataset): |
| 649 | """ |
| 650 | WN18 knowledge graph dataset. |
| 651 | |
| 652 | Splits: |
| 653 | train, valid, test |
| 654 | """ |
| 655 | def __init__(self): |
| 656 | super(WN18, self).__init__( |
| 657 | "wn18", |
| 658 | urls={ |
| 659 | "train": "https://github.com/DeepGraphLearning/KnowledgeGraphEmbedding/raw/master/data/wn18/train.txt", |
| 660 | "valid": "https://github.com/DeepGraphLearning/KnowledgeGraphEmbedding/raw/master/data/wn18/valid.txt", |
| 661 | "test": "https://github.com/DeepGraphLearning/KnowledgeGraphEmbedding/raw/master/data/wn18/test.txt" |
| 662 | } |
| 663 | ) |
| 664 | |
| 665 | |
| 666 | class WN18RR(Dataset): |