Skip to content

Commit 66744a1

Browse files
committed
repair createAliasTable
1 parent 575ae26 commit 66744a1

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

Graph Computing.xmind

-9.19 KB
Binary file not shown.

src/main/java/com/gnn/embedding/Struc2vec.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -244,10 +244,10 @@ public void buildWeightedLayeredGraph(Map<Pair<K, K>, List<Double>> pairDistance
244244
if (layersAdj.size() <= i) {
245245
layersAdj.add(new HashMap<>());
246246
}
247-
if (layersAdj.get(i).containsKey(pair.getKey())) {
247+
if (!layersAdj.get(i).containsKey(pair.getKey())) {
248248
layersAdj.get(i).put(pair.getKey(), new ArrayList<>());
249249
}
250-
if (layersAdj.get(i).containsKey(pair.getValue())) {
250+
if (!layersAdj.get(i).containsKey(pair.getValue())) {
251251
layersAdj.get(i).put(pair.getValue(), new ArrayList<>());
252252
}
253253
layersAdj.get(i).get(pair.getKey()).add(pair.getValue());
@@ -301,6 +301,8 @@ public void createAliasTable(List<Double> edgeWeight, K v, Map<K, List<Double>>
301301
Stack<Integer> large = new Stack<>();
302302
List<Double> edgeWeight_ = new ArrayList<>();
303303
for (int i = 0; i < edgeWeight.size(); ++i) {
304+
accept.add(0.0);
305+
alias.add(0.0);
304306
edgeWeight_.add(edgeWeight.get(i) * edgeWeight.size());
305307
if (edgeWeight_.get(i) < 1) {
306308
small.push(i);

0 commit comments

Comments
 (0)