From 1009ef3ccecfeae037675bae90636f420e2d65c0 Mon Sep 17 00:00:00 2001 From: Rajdeep Dua Date: Sun, 15 May 2022 09:47:35 +0530 Subject: [PATCH] updated to latest version of libraries --- requirements.txt | 6 +++--- src/main.py | 9 +++++++-- src/node2vec.py | 4 ++-- 3 files changed, 12 insertions(+), 7 deletions(-) diff --git a/requirements.txt b/requirements.txt index faa710ca..c89a2230 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,3 @@ -networkx==1.11 -numpy==1.11.2 -gensim==0.13.3 +networkx==2.8 +numpy>=1.22.3 +gensim==4.2.0 diff --git a/src/main.py b/src/main.py index 82ac7357..d988d7da 100644 --- a/src/main.py +++ b/src/main.py @@ -84,8 +84,13 @@ def learn_embeddings(walks): Learn embeddings by optimizing the Skipgram objective using SGD. ''' walks = [map(str, walk) for walk in walks] - model = Word2Vec(walks, size=args.dimensions, window=args.window_size, min_count=0, sg=1, workers=args.workers, iter=args.iter) - model.save_word2vec_format(args.output) + # size=args.dimensions + model = Word2Vec(walks, + vector_size=args.dimensions, + window=args.window_size, min_count=0, sg=1, workers=args.workers, #iter=args.iter + ) + #model.save_word2vec_format(args.output) + model.save(args.output) return diff --git a/src/node2vec.py b/src/node2vec.py index 0293411a..66ff6540 100644 --- a/src/node2vec.py +++ b/src/node2vec.py @@ -43,9 +43,9 @@ def simulate_walks(self, num_walks, walk_length): G = self.G walks = [] nodes = list(G.nodes()) - print 'Walk iteration:' + print('Walk iteration:') for walk_iter in range(num_walks): - print str(walk_iter+1), '/', str(num_walks) + print(str(walk_iter+1), '/', str(num_walks)) random.shuffle(nodes) for node in nodes: walks.append(self.node2vec_walk(walk_length=walk_length, start_node=node))