Update core.py

Redundant arithmetic fix
xiaoouwang · Aug 31, 2020 · 4257a38 · 4257a38
1 parent 8b93d75
commit 4257a38
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)