Class FB15k237

python/graphvite/dataset.py:630–645

FB15k-237 knowledge graph dataset. Splits: train, valid, test

class FB15k237(Dataset):
    """
    FB15k-237 knowledge graph dataset.

    Splits:
        train, valid, test
    """
    def __init__(self):
        super(FB15k237, self).__init__(
            "fb15k-237",
            urls={
                "train": "https://github.com/DeepGraphLearning/KnowledgeGraphEmbedding/raw/master/data/FB15k-237/train.txt",
                "valid": "https://github.com/DeepGraphLearning/KnowledgeGraphEmbedding/raw/master/data/FB15k-237/valid.txt",
                "test": "https://github.com/DeepGraphLearning/KnowledgeGraphEmbedding/raw/master/data/FB15k-237/test.txt"
            }
        )


class WN18(Dataset):