Merge pull request #55 from kkteru/sg_default

Fix for issue #54
eliorc · Jan 9, 2021 · 361e33b · 361e33b
2 parents 8f73ca5 + 628ddb3
commit 361e33b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/node2vec/node2vec.py b/node2vec/node2vec.py
@@ -177,4 +177,7 @@ def fit(self, **skip_gram_params) -> gensim.models.Word2Vec:
         if 'size' not in skip_gram_params:
             skip_gram_params['size'] = self.dimensions
 
+        if 'sg' not in skip_gram_params:
+            skip_gram_params['sg'] = 1
+
         return gensim.models.Word2Vec(self.walks, **skip_gram_params)