MCPcopy Create free account
hub / github.com/Vegetebird/GraphMLP / basic_layout

Method basic_layout

model/block/graph_frames.py:36–53  ·  view source on GitHub ↗
(self,neighbour_base, sym_base)

Source from the content-addressed store, hash-verified

34 return [((front - 1) + i*self.num_node_each, (back - 1)+ i*self.num_node_each) for i in range(self.seqlen) for (front, back) in base]
35
36 def basic_layout(self,neighbour_base, sym_base):
37 self.num_node = self.num_node_each * self.seqlen
38 time_link = [(i * self.num_node_each + j, (i + 1) * self.num_node_each + j) for i in range(self.seqlen - 1)
39 for j in range(self.num_node_each)]
40 self.time_link_forward = [(i * self.num_node_each + j, (i + 1) * self.num_node_each + j) for i in
41 range(self.seqlen - 1)
42 for j in range(self.num_node_each)]
43 self.time_link_back = [((i + 1) * self.num_node_each + j, (i) * self.num_node_each + j) for i in
44 range(self.seqlen - 1)
45 for j in range(self.num_node_each)]
46
47 self_link = [(i, i) for i in range(self.num_node)]
48
49 self.neighbour_link_all = self.graph_link_between_frames(neighbour_base)
50
51 self.sym_link_all = self.graph_link_between_frames(sym_base)
52
53 return self_link, time_link
54
55 def get_edge(self, layout):
56 if layout == 'hm36_gt':

Callers 1

get_edgeMethod · 0.95

Calls 1

Tested by

no test coverage detected