aditya-grover / node2vec

http://snap.stanford.edu/node2vec/
MIT License
2.61k stars 912 forks source link

Change args passed to Word2Vec class #111

Open pedroruas18 opened 3 years ago

pedroruas18 commented 3 years ago

The args "size" and "iter" (line 87, in the script "src/main.py") are no longer used in Gensim.4.0 (see https://github.com/RaRe-Technologies/gensim/wiki/Migrating-from-Gensim-3.x-to-4), which throws an error, so I changed them to "vector_size" and "epochs", respectively.