MCPcopy Create free account
hub / github.com/FastMAS/KVCOMM / generate_layered_graph

Function generate_layered_graph

experiments/run_mmlu.py:153–167  ·  view source on GitHub ↗
(n, layer_num=2)

Source from the content-addressed store, hash-verified

151 node_kwargs = None
152
153 def generate_layered_graph(n, layer_num=2):
154 adj_matrix = [[0] * n for _ in range(n)]
155 base_size = n // layer_num
156 remainder = n % layer_num
157 layers: List[int] = []
158 for i in range(layer_num):
159 size = base_size + (1 if i < remainder else 0)
160 layers.extend([i] * size)
161 random.shuffle(layers)
162 for i in range(n):
163 current_layer = layers[i]
164 for j in range(n):
165 if layers[j] == current_layer + 1:
166 adj_matrix[i][j] = 1
167 return adj_matrix
168
169 def generate_mesh_graph(n):
170 adj_matrix = [[0] * n for _ in range(n)]

Callers 1

get_kwargsFunction · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected