(self)
| 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): |