From 5a7dab7685315aa7bb94ac0b8cec7bb39f7f0e0a Mon Sep 17 00:00:00 2001 From: Evgeny Bazarov Date: Mon, 20 Aug 2018 16:46:38 +0200 Subject: [PATCH] fix pretrain argument fix pretrain argument which will store now False when no activated, when before it was always true --- src/han/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/han/main.py b/src/han/main.py index 1201dcd..ab96dee 100644 --- a/src/han/main.py +++ b/src/han/main.py @@ -38,7 +38,7 @@ def get_args(): parser.add_argument("--dataset", type=str, default='imdb') parser.add_argument("--data_folder", type=str, default="datasets/imdb/han") parser.add_argument("--model_folder", type=str, default="models/han/imdb") - parser.add_argument("--pretrain",action="store_true", default=True, help="pre train embeddings with word2vec (gensim required)") + parser.add_argument("--pretrain",action="store_true", default=False, help="pre train embeddings with word2vec (gensim required)") parser.add_argument('--max_feats', type=int, default=10000, help="vocabulary size") parser.add_argument('--max_sents', type=int, default=-1, help="max number of sentences per example") parser.add_argument('--max_words', type=int, default=-1, help="max word length")