| 689 | train, valid, test |
| 690 | """ |
| 691 | def __init__(self): |
| 692 | super(Wikidata5m, self).__init__( |
| 693 | "wikidata5m", |
| 694 | urls={ |
| 695 | "train": "https://www.dropbox.com/s/dty6ufe1gg6keuc/wikidata5m.txt.gz?dl=1", |
| 696 | "valid": "https://www.dropbox.com/s/dty6ufe1gg6keuc/wikidata5m.txt.gz?dl=1", |
| 697 | "test": "https://www.dropbox.com/s/dty6ufe1gg6keuc/wikidata5m.txt.gz?dl=1", |
| 698 | "entity": "https://www.dropbox.com/s/bgmgvk8brjwpc9w/entity.txt.gz?dl=1", |
| 699 | "relation": "https://www.dropbox.com/s/37jxki93gguv0pp/relation.txt.gz?dl=1", |
| 700 | "alias2entity": [], # depends on `entity` |
| 701 | "alias2relation": [] # depends on `relation` |
| 702 | } |
| 703 | ) |
| 704 | |
| 705 | def train_preprocess(self, graph_file, train_file): |
| 706 | valid_file = train_file[:train_file.rfind("train.txt")] + "valid.txt" |