Merge pull request #54 from zhengjxu/patch-1

Fix bug in oneRandomWalk
wzhe06 · Sep 9, 2021 · 579577c · 579577c
2 parents 90d20f8 + 86bc669
commit 579577c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/main/java/com/sparrowrecsys/offline/spark/embedding/Embedding.scala b/src/main/java/com/sparrowrecsys/offline/spark/embedding/Embedding.scala
@@ -162,8 +162,10 @@ object Embedding {
 
       val probDistribution = transitionMatrix(curElement)
       val randomDouble = Random.nextDouble()
+      var accumulateProb: Double = 0D
       breakable { for ((item, prob) <- probDistribution) {
-        if (randomDouble >= prob){
+        accumulateProb += prob
+        if (accumulateProb >= randomDouble){
           curElement = item
           break
         }