search

saakur / MG2V

Official code for the MG2V framework
MIT License
0 stars 0 forks source link

Word2Vec: size to vector_size #2

Open WilliamTun opened 1 year ago

WilliamTun commented 1 year ago

line 87 n2v_main.py:

model = Word2Vec(walks, size=args.dimensions, window=args.window_size, min_count=0, sg=1, workers=args.workers, iter=args.iter)

The "size" argument needs to be changed to "vector_size":

model = Word2Vec(walks, vector_size=args.dimensions, window=args.window_size, min_count=0, sg=1, workers=args.workers, iter=args.iter)

reference: https://stackoverflow.com/questions/53195906/getting-init-got-an-unexpected-keyword-argument-document-this-error-in

WilliamTun commented 1 year ago

On the same line, please change "iter" to "epochs" too!