(self, graph, walk_length, num_walks, p=1.0, q=1.0, workers=1, use_rejection_sampling=False)
| 26 | class Node2Vec: |
| 27 | |
| 28 | def __init__(self, graph, walk_length, num_walks, p=1.0, q=1.0, workers=1, use_rejection_sampling=False): |
| 29 | |
| 30 | self.graph = graph |
| 31 | self._embeddings = {} |
| 32 | self.walker = RandomWalker( |
| 33 | graph, p=p, q=q, use_rejection_sampling=use_rejection_sampling) |
| 34 | |
| 35 | print("Preprocess transition probs...") |
| 36 | self.walker.preprocess_transition_probs() |
| 37 | |
| 38 | self.sentences = self.walker.simulate_walks( |
| 39 | num_walks=num_walks, walk_length=walk_length, workers=workers, verbose=1) |
| 40 | |
| 41 | def train(self, embed_size=128, window_size=5, workers=3, iter=5, **kwargs): |
| 42 | kwargs["sentences"] = self.sentences |
nothing calls this directly
no test coverage detected