chore(nalp): Changes support for gensim v4+.

gugarosa · Jan 19, 2022 · f90b2a0 · f90b2a0
1 parent 8a7d8b7
commit f90b2a0
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git a/nalp/encoders/word2vec.py b/nalp/encoders/word2vec.py
@@ -46,8 +46,8 @@ def learn(self, tokens, max_features=128, window_size=5, min_count=1,
 
         """
 
-        self.encoder = W2V(sentences=[tokens], size=max_features, window=window_size, min_count=min_count,
-                           sg=algorithm, alpha=learning_rate, iter=iterations,
+        self.encoder = W2V(sentences=[tokens], vector_size=max_features, window=window_size, min_count=min_count,
+                           sg=algorithm, alpha=learning_rate, epochs=iterations,
                            workers=multiprocessing.cpu_count())
 
     def encode(self, tokens):
@@ -94,7 +94,6 @@ def decode(self, encoded_tokens):
 
             raise RuntimeError(e)
 
-        decoded_tokens = [self.encoder.wv.most_similar(
-            positive=[t])[0][0] for t in encoded_tokens]
+        decoded_tokens = [self.encoder.wv.most_similar(positive=[t])[0][0] for t in encoded_tokens]
 
         return decoded_tokens
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 coverage>=5.5
-gensim>=3.8.3
+gensim>=4.1.2
 matplotlib>=3.3.4
 mido>=1.2.9
 nltk>=3.5

diff --git a/setup.py b/setup.py
@@ -13,7 +13,7 @@
       url='https://github.com/gugarosa/nalp',
       license='Apache 2.0',
       install_requires=['coverage>=5.5',
-                        'gensim>=3.8.3',
+                        'gensim>=4.1.2',
                         'matplotlib>=3.3.4',
                         'mido>=1.2.9',
                         'nltk>=3.5',