Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

word2vec parameters changed #49

Closed
daniel-unyi-42 opened this issue Feb 17, 2022 · 3 comments
Closed

word2vec parameters changed #49

daniel-unyi-42 opened this issue Feb 17, 2022 · 3 comments

Comments

@daniel-unyi-42
Copy link

Hi!

In node2vec.py, you should modify the 'iter' parameter to 'epochs' and the 'size' parameter to 'vector_size'.

(And thank you for the library, I use it extensively in my research!)

@VHRanger
Copy link
Owner

The parameters in the Node2Vec class are to stay consistent with SKLearn conventions (n_components and epochs)

Do you mean the ones passed to gensim are erroneous?

@daniel-unyi-42
Copy link
Author

Yes, the parameter names in the constructor of gensim.models.Word2Vec were changed.
Sorry for the misunderstanding.

@VHRanger
Copy link
Owner

Ah, actually it's tracked here:

#37

I'll close this one and resolve the other one right away

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants