Merge pull request #40 from radkoff/master

fix KeyError during node2vec
2020-07-25 20:30:37 +08:00 · 2020-07-25 20:30:37 +08:00 · 6f4aac5e42
commit 6f4aac5e42
parent 7de7a09aa1 dc135a8475
1 changed files with 2 additions and 0 deletions
--- a/ge/walker.py
+++ b/ge/walker.py
@ -191,6 +191,8 @@ class RandomWalker:

            for edge in G.edges():
                alias_edges[edge] = self.get_alias_edge(edge[0], edge[1])
+                if not G.is_directed():
+                    alias_edges[(edge[1], edge[0])] = self.get_alias_edge(edge[1], edge[0])
                self.alias_edges = alias_edges

        self.alias_nodes = alias_nodes