(self, num_walks, walk_length)
| 126 | return walk |
| 127 | |
| 128 | def _simulate_walks(self, num_walks, walk_length): |
| 129 | # Repeatedly simulate random walks from each node. |
| 130 | G = self.G |
| 131 | walks = [] |
| 132 | nodes = list(G.nodes()) |
| 133 | print("Walk iteration:") |
| 134 | for walk_iter in range(num_walks): |
| 135 | if walk_iter % 10 == 0: |
| 136 | print(str(walk_iter + 1), "/", str(num_walks)) |
| 137 | random.shuffle(nodes) |
| 138 | for node in nodes: |
| 139 | walks.append(self._node2vec_walk(walk_length=walk_length, start_node=node)) |
| 140 | |
| 141 | return walks |
| 142 | |
| 143 | def _get_alias_edge(self, src, dst): |
| 144 | # Get the alias edge setup lists for a given edge. |
no test coverage detected