dmlc--dgl
Lingfan Yu
0f121eeb76
* rgcn tutorial * data processing for rgcn * many fix * requirements for rgcn tutorial * fix comments * description of rgcn dataset * author * bug fix * move all dataset to s3
11 行
415 B
Python
11 行
415 B
Python
from __future__ import absolute_import
|
|
|
|
from . import knowledge_graph as knwlgrh
|
|
def load_data(dataset, bfs_level=3, relabel=False):
|
|
if dataset in ['aifb', 'mutag', 'bgs', 'am']:
|
|
return knwlgrh.load_entity(dataset, bfs_level, relabel)
|
|
elif dataset in ['FB15k', 'wn18', 'FB15k-237']:
|
|
return knwlgrh.load_link(dataset)
|
|
else:
|
|
raise ValueError('Unknown dataset: {}'.format(dataset))
|