Skip to content

Instantly share code, notes, and snippets.

@prateekjoshi565

prateekjoshi565/lp_train_node2vec.py

Created December 29, 2019 07:44
Show Gist options
  • Save prateekjoshi565/45cea3efb5ccd278c9304ebf83c7adb3 to your computer and use it in GitHub Desktop.
Save prateekjoshi565/45cea3efb5ccd278c9304ebf83c7adb3 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
lp_train_node2vec.py
from node2vec import Node2Vec
# Generate walks
node2vec = Node2Vec(G_data, dimensions=100, walk_length=16, num_walks=50)
# train node2vec model
n2w_model = node2vec.fit(window=7, min_count=1)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment