From 4257a385dacb9b721d32dcd6987b6f8bb749e609 Mon Sep 17 00:00:00 2001
From: KirillErofeev <krllerof@gmail.com>
Date: Mon, 31 Aug 2020 22:35:58 +0300
Subject: [PATCH] Update core.py

Redundant arithmetic fix
---
 svd2vec/core.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/svd2vec/core.py b/svd2vec/core.py
index f43905d..163c9eb 100644
--- 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)