def setup_fixed(): nodes = {} for i in range(8): node = g.Node(i) nodes[i] = node conns = { 0: {1, 2}, 1: {0, 2}, 2: {0, 1, 3, 7}, 3: {2, 4, 6}, 4: {3}, 5: {2}, 6: {3}, 7: {2} } edges = {} i = 0 for nid1 in conns: for nid2 in conns[nid1]: edge = g.Edge(i, nodes[nid1], nodes[nid2]) edges[i] = edge i += 1 return g.Graph(edges)
def setup_random(num_nodes, num_edges, multi=False): nodes = {} for i in range(num_nodes): node = g.Node(i) nodes[i] = node conns = {} edges = {} for i in range(num_edges): nid1, nid2 = random.sample(np.arange(num_nodes).tolist(), 2) if not multi: if (nid1, nid2) in conns or (nid2, nid1) in conns: continue edge = g.Edge(i, nodes[nid1], nodes[nid2], data=util.compute_view_number(nodes[nid1], nodes[nid2])) conns[(nid1, nid2)] = i edges[i] = edge return g.Graph(edges)
def simple_interpret_edge(eid, node1, node2, tokens): return g.Edge(eid, node1, node2)