diff options
| author | schneefux <schneefux+commit@schneefux.xyz> | 2018-04-03 12:37:46 +0200 |
|---|---|---|
| committer | schneefux <schneefux+commit@schneefux.xyz> | 2018-04-03 12:37:46 +0200 |
| commit | 6a5f362ccced3f2c6704c22712de0d6ceb9a5a4d (patch) | |
| tree | e0c4faa954cf3c9090f928662f30322ba9f12edc /processors/redis_to_word2vec.py | |
| parent | 2a20179c47c7a28c8cca05e1913126f95886197c (diff) | |
| download | ragequitte-6a5f362ccced3f2c6704c22712de0d6ceb9a5a4d.tar.gz ragequitte-6a5f362ccced3f2c6704c22712de0d6ceb9a5a4d.zip | |
improve logging
Diffstat (limited to 'processors/redis_to_word2vec.py')
| -rw-r--r-- | processors/redis_to_word2vec.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/processors/redis_to_word2vec.py b/processors/redis_to_word2vec.py index 752ee37..fd5f521 100644 --- a/processors/redis_to_word2vec.py +++ b/processors/redis_to_word2vec.py @@ -97,12 +97,12 @@ class RedisIterator(object): def discord_batch_handler(message): global model_updating - print("building vocabulary") + logging.info("building vocabulary") model.build_vocab(RedisIterator(match="message:*", deleting=False), update=model_updating) - print("vocabulary size: {}".format(len(model.wv.vocab))) - print("training") + logging.info("vocabulary size: {}".format(len(model.wv.vocab))) + logging.info("training") model.train(RedisIterator(match="message:*", deleting=True), total_examples=model.corpus_count, epochs=model.epochs) - print("trained") + logging.info("trained") model.save(config.model) model_updating = True if config.debug: |