avoid zero

apache · Feb 22, 2022 · 3fea9a5 · 3fea9a5
1 parent 2fc2cc9
commit 3fea9a5
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 4 deletions.
diff --git a/hugegraph-api/src/main/java/com/baidu/hugegraph/api/traversers/AdamicAdarAPI.java b/hugegraph-api/src/main/java/com/baidu/hugegraph/api/traversers/AdamicAdarAPI.java
@@ -46,7 +46,8 @@
 
 /**
  * AdamicAdar is one of the prediction algorithms in graph, you can get more
- * info and definition in wikipedia
+ * info and definition in:
+ * https://en.wikipedia.org/wiki/Adamic/Adar_index
  */
 @Path("graphs/{graph}/traversers/adamicadar")
 @Singleton

diff --git a/hugegraph-api/src/main/java/com/baidu/hugegraph/api/traversers/ResourceAllocationAPI.java b/hugegraph-api/src/main/java/com/baidu/hugegraph/api/traversers/ResourceAllocationAPI.java
@@ -46,7 +46,8 @@
 
 /**
  * ResourceAllocation is one of the prediction algorithms in graph, you can get
- * more info and definition in wikipedia
+ * more info and definition in:
+ * https://arxiv.org/pdf/0901.0553.pdf
  */
 @Path("graphs/{graph}/traversers/resourceallocation")
 @Singleton

diff --git a/...graph-core/src/main/java/com/baidu/hugegraph/traversal/algorithm/PredictionTraverser.java b/...graph-core/src/main/java/com/baidu/hugegraph/traversal/algorithm/PredictionTraverser.java
@@ -43,7 +43,10 @@ public double adamicAdar(Id source, Id target, Directions dir,
 
         double sum = 0.0;
         for (Id vid : neighbors) {
-            sum += 1.0 / Math.log(this.edgesCount(vid, step));
+            long currentDegree = this.edgesCount(vid, step);
+            if (currentDegree > 0) {
+                sum += 1.0 / Math.log(currentDegree);
+            }
         }
         return sum;
     }
@@ -57,7 +60,10 @@ public double resourceAllocation(Id source, Id target, Directions dir,
 
         double sum = 0.0;
         for (Id vid : neighbors) {
-            sum += 1.0 / this.edgesCount(vid, step);
+            long currentDegree = this.edgesCount(vid, step);
+            if (currentDegree > 0) {
+                sum += 1.0 / currentDegree;
+            }
         }
         return sum;
     }