(self, graph, walk_length, num_walks, workers=1)
| 24 | |
| 25 | class DeepWalk: |
| 26 | def __init__(self, graph, walk_length, num_walks, workers=1): |
| 27 | |
| 28 | self.graph = graph |
| 29 | self.w2v_model = None |
| 30 | self._embeddings = {} |
| 31 | |
| 32 | self.walker = RandomWalker( |
| 33 | graph, p=1, q=1, ) |
| 34 | self.sentences = self.walker.simulate_walks( |
| 35 | num_walks=num_walks, walk_length=walk_length, workers=workers, verbose=1) |
| 36 | |
| 37 | def train(self, embed_size=128, window_size=5, workers=3, iter=5, **kwargs): |
| 38 |
nothing calls this directly
no test coverage detected