Skip to content

GitLab

Unverified Commit f9ad1c80 authored by Da Zheng's avatar Da Zheng Committed by GitHub
Browse files

fix. (#1358) 


Co-authored-by: default avatarChao Ma <mctt90@gmail.com>
parent eb71c804
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
apps/kg/dataloader/sampler.py
View file @ f9ad1c80
......@@ -275,7 +275,7 @@ def ConstructGraph(edges, n_entities, args):
else:
src, etype_id, dst = edges
coo = sp.sparse.coo_matrix((np.ones(len(src)), (src, dst)), shape=[n_entities, n_entities])
g = dgl.DGLGraph(coo, readonly=True, sort_csr=True)
g = dgl.DGLGraph(coo, readonly=True, multigraph=True, sort_csr=True)
g.edata['tid'] = F.tensor(etype_id, F.int64)
if args.pickle_graph:
with open(os.path.join(args.data_path, args.dataset, pickle_name), 'wb') as graph_file:
......@@ -562,7 +562,7 @@ class EvalDataset(object):
dst = np.concatenate((dataset.train[2], dataset.valid[2], dataset.test[2]))
coo = sp.sparse.coo_matrix((np.ones(len(src)), (src, dst)),
shape=[dataset.n_entities, dataset.n_entities])
g = dgl.DGLGraph(coo, readonly=True, sort_csr=True)
g = dgl.DGLGraph(coo, readonly=True, multigraph=True, sort_csr=True)
g.edata['tid'] = F.tensor(etype_id, F.int64)
if args.pickle_graph:
with open(os.path.join(args.data_path, args.dataset, pickle_name), 'wb') as graph_file:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment