Merge pull request valentinp72#2 from KirillErofeev/patch-1

Redundant arithmetic in core.py
xiaoouwang · Nov 29, 2020 · c97998d · c97998d
2 parents 8b93d75 + 4257a38
commit c97998d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/svd2vec/core.py b/svd2vec/core.py
@@ -196,8 +196,8 @@ def subsampling(self):
                 if self.terms_counts[word] < self.min_count:
                     continue
                 word_frequency = 1.0 * self.terms_counts[word] / self.d_size
-                prob = 1 - np.sqrt(self.sub_threshold / word_frequency)
-                if not Utils.random_decision(prob):
+                prob = np.sqrt(self.sub_threshold / word_frequency)
+                if Utils.random_decision(prob):
                     # we keep the word
                     new_words.append(word)
             new_docs.append(new_words)