From 25af18b37e10c8e858529f49de39fc30cf46ae2e Mon Sep 17 00:00:00 2001 From: Yoel Zeldes Date: Tue, 16 Oct 2018 23:30:46 +0300 Subject: [PATCH] bug fix: model.save_word2vec_format is deprecated --- src/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/main.py b/src/main.py index 82ac735..ea08532 100644 --- a/src/main.py +++ b/src/main.py @@ -85,7 +85,7 @@ def learn_embeddings(walks): ''' 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) + model.wv.save_word2vec_format(args.output) return