From 6bcf1885ed3428313c24563280f4f1198194abe5 Mon Sep 17 00:00:00 2001
From: Slava Vishnyakov <bomboze@gmail.com>
Date: Sat, 21 Apr 2018 13:25:50 +0300
Subject: [PATCH] Fix algorithm

---
 src/NlpTools/Models/Lda.php | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/NlpTools/Models/Lda.php b/src/NlpTools/Models/Lda.php
index bec01d2..efde3cf 100644
--- a/src/NlpTools/Models/Lda.php
+++ b/src/NlpTools/Models/Lda.php
@@ -240,7 +240,7 @@ public function getDocumentsPerTopicsProbabilities($limit_docs=-1)
          $count_topics_docs = array();
          foreach ($this->count_docs_topics as $doc=>$topics) {
              foreach ($topics as $t=>$c)
-                $count_topics_docs[$doc][$t]++;
+                $count_topics_docs[$doc][$t]+=$c;
          }
 
          foreach ($p_t_d as $topic=>&$p) {
@@ -253,7 +253,7 @@ public function getDocumentsPerTopicsProbabilities($limit_docs=-1)
              }
          }
 
-         return $p;
+         return $p_t_d;
      }
 
      /**