diff --git a/BioWordVec.py b/BioWordVec.py index 806ed6b..749f301 100644 --- a/BioWordVec.py +++ b/BioWordVec.py @@ -115,8 +115,8 @@ def main(args): temp_list.append(mesh_dict[node]) new_walks.append(temp_list) - model = FastText(MySentences(new_walks,args.input_corpus), size=args.dimensions, window=args.windows, min_count=args.min_count, workers=args.workers, - sg=args.sg, iter=args.iter) + model = FastText(MySentences(new_walks,args.input_corpus), window=args.windows, min_count=args.min_count, workers=args.workers, + sg=args.sg) # size=args.dimensions, , iter=args.iter model.save(args.output_model) diff --git a/node2vec.py b/node2vec.py index affeee4..a10b374 100644 --- a/node2vec.py +++ b/node2vec.py @@ -119,7 +119,7 @@ def alias_setup(probs): ''' K = len(probs) q = np.zeros(K) - J = np.zeros(K, dtype=np.int) + J = np.zeros(K, dtype=int) smaller = [] larger = [] diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..4790de5 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,4 @@ +networkx +node2vec +gensim +numpy