Skip to content

Commit 361e33b

Browse files
authored
Merge pull request #55 from kkteru/sg_default
Fix for issue #54
2 parents 8f73ca5 + 628ddb3 commit 361e33b

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

node2vec/node2vec.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -177,4 +177,7 @@ def fit(self, **skip_gram_params) -> gensim.models.Word2Vec:
177177
if 'size' not in skip_gram_params:
178178
skip_gram_params['size'] = self.dimensions
179179

180+
if 'sg' not in skip_gram_params:
181+
skip_gram_params['sg'] = 1
182+
180183
return gensim.models.Word2Vec(self.walks, **skip_gram_params)

0 commit comments

Comments
 (0)