From cc6f125b8c4437af5588bb679ca2af18e6248a99 Mon Sep 17 00:00:00 2001 From: schneefux Date: Tue, 3 Apr 2018 21:38:08 +0200 Subject: word2vec model: use cbow --- processors/redis_to_word2vec.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'processors') diff --git a/processors/redis_to_word2vec.py b/processors/redis_to_word2vec.py index 787f48c..b3de703 100644 --- a/processors/redis_to_word2vec.py +++ b/processors/redis_to_word2vec.py @@ -19,7 +19,7 @@ model_updating = True try: model = gensim.models.Word2Vec.load(config.model) except Exception as e: - model = gensim.models.Word2Vec(iter=1, min_count=10, size=200, workers=4, sg=1) + model = gensim.models.Word2Vec(iter=1, min_count=10, size=200, workers=4, sg=0) # defaults. iter=1: only iterate once, sg=1: use skip gram (needs more data) model_updating = False -- cgit v1.3.1