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)