MCPcopy
hub / github.com/shenweichen/GraphEmbedding / simulate_walks

Method simulate_walks

ge/walker.py:208–223  ·  view source on GitHub ↗
(self, num_walks, walk_length, stay_prob=0.3, workers=1, verbose=0)

Source from the content-addressed store, hash-verified

206 pass
207
208 def simulate_walks(self, num_walks, walk_length, stay_prob=0.3, workers=1, verbose=0):
209
210 layers_adj = pd.read_pickle(self.temp_path + 'layers_adj.pkl')
211 layers_alias = pd.read_pickle(self.temp_path + 'layers_alias.pkl')
212 layers_accept = pd.read_pickle(self.temp_path + 'layers_accept.pkl')
213 gamma = pd.read_pickle(self.temp_path + 'gamma.pkl')
214
215 nodes = self.idx # list(self.g.nodes())
216
217 results = Parallel(n_jobs=workers, verbose=verbose, )(
218 delayed(self._simulate_walks)(nodes, num, walk_length, stay_prob, layers_adj, layers_accept, layers_alias,
219 gamma) for num in
220 partition_num(num_walks, workers))
221
222 walks = list(itertools.chain(*results))
223 return walks
224
225 def _simulate_walks(self, nodes, num_walks, walk_length, stay_prob, layers_adj, layers_accept, layers_alias, gamma):
226 walks = []

Callers

nothing calls this directly

Calls 1

partition_numFunction · 0.85

Tested by

no test coverage detected