diff options
| author | schneefux <schneefux+commit@schneefux.xyz> | 2018-04-03 21:38:08 +0200 |
|---|---|---|
| committer | schneefux <schneefux+commit@schneefux.xyz> | 2018-04-03 21:38:08 +0200 |
| commit | cc6f125b8c4437af5588bb679ca2af18e6248a99 (patch) | |
| tree | 48e0db41829129d7d945272266d2b4cef4ef60d2 | |
| parent | dbff41e0f42317cb8708bb3b6298ea90cd946f9a (diff) | |
| download | ragequitte-cc6f125b8c4437af5588bb679ca2af18e6248a99.tar.gz ragequitte-cc6f125b8c4437af5588bb679ca2af18e6248a99.zip | |
word2vec model: use cbow
| -rw-r--r-- | processors/redis_to_word2vec.py | 2 |
1 files changed, 1 insertions, 1 deletions
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 |