[Doc] RGCN tutorial (dmlc#206)

* 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
cnavasca95 · Dec 1, 2018 · 0f121ee · 0f121ee
1 parent cccc233
commit 0f121ee
Show file tree

Hide file tree

Showing 4 changed files with 906 additions and 0 deletions.
diff --git a/python/dgl/contrib/data/__init__.py b/python/dgl/contrib/data/__init__.py
@@ -0,0 +1,10 @@
+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))