Fixed the problem with commit "unnecessary importing for gsdmm and bt…

…m" (#50)
fani-lab · Oct 10, 2022 · 3788289 · 3788289
1 parent dc2f830
commit 3788289
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/uml/UserSimilarities.py b/src/uml/UserSimilarities.py
@@ -30,7 +30,7 @@ def main(documents, dictionary, lda_model, path2_save_uml, just_one, binary, thr
                 doc = dp.tagme_annotator(row['Text'])
             else:
                 doc = row['Text']
-            if Params.tml['method'].lower() == 'btm': d2t = tm.doc2topics(lda_model, doc, threshold=threshold, just_one=just_one, binary=binary, dict=dictionary)
+            if Params.tml['method'].lower() == 'btm': d2t = tm.doc2topics(lda_model, [doc], threshold=Params.evl['threshold'], just_one=Params.tml['justOne'], binary=Params.tml['binary'], dic=dictionary)
             else: d2t = tm.doc2topics(lda_model, dictionary.doc2bow(doc.split()), threshold=threshold, just_one=just_one, binary=binary)
             users_topic_interests[row['UserId']] = d2t