Unverified Commit 337b1559 authored by Zhou Fang's avatar Zhou Fang Committed by GitHub
Browse files

[Doc] fix tutorial - blitz 4 link prediction (#2726)



The number of edges should be same as posive eids.
Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
parent 09ef2c2c
...@@ -97,7 +97,7 @@ adj = sp.coo_matrix((np.ones(len(u)), (u.numpy(), v.numpy()))) ...@@ -97,7 +97,7 @@ adj = sp.coo_matrix((np.ones(len(u)), (u.numpy(), v.numpy())))
adj_neg = 1 - adj.todense() - np.eye(g.number_of_nodes()) adj_neg = 1 - adj.todense() - np.eye(g.number_of_nodes())
neg_u, neg_v = np.where(adj_neg != 0) neg_u, neg_v = np.where(adj_neg != 0)
neg_eids = np.random.choice(len(neg_u), g.number_of_edges() // 2) neg_eids = np.random.choice(len(neg_u), g.number_of_edges())
test_neg_u, test_neg_v = neg_u[neg_eids[:test_size]], neg_v[neg_eids[:test_size]] test_neg_u, test_neg_v = neg_u[neg_eids[:test_size]], neg_v[neg_eids[:test_size]]
train_neg_u, train_neg_v = neg_u[neg_eids[test_size:]], neg_v[neg_eids[test_size:]] train_neg_u, train_neg_v = neg_u[neg_eids[test_size:]], neg_v[neg_eids[test_size:]]
......
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