FB15k knowledge graph dataset. Splits: train, valid, test
| 610 | |
| 611 | |
| 612 | class FB15k(Dataset): |
| 613 | """ |
| 614 | FB15k knowledge graph dataset. |
| 615 | |
| 616 | Splits: |
| 617 | train, valid, test |
| 618 | """ |
| 619 | def __init__(self): |
| 620 | super(FB15k, self).__init__( |
| 621 | "fb15k", |
| 622 | urls={ |
| 623 | "train": "https://github.com/DeepGraphLearning/KnowledgeGraphEmbedding/raw/master/data/FB15k/train.txt", |
| 624 | "valid": "https://github.com/DeepGraphLearning/KnowledgeGraphEmbedding/raw/master/data/FB15k/valid.txt", |
| 625 | "test": "https://github.com/DeepGraphLearning/KnowledgeGraphEmbedding/raw/master/data/FB15k/test.txt" |
| 626 | } |
| 627 | ) |
| 628 | |
| 629 | |
| 630 | class FB15k237(Dataset): |