diff options
| author | Roman Grundkiewicz <rgrundki@exseed.ed.ac.uk> | 2017-07-02 09:56:38 +0300 |
|---|---|---|
| committer | Roman Grundkiewicz <rgrundki@exseed.ed.ac.uk> | 2017-07-02 09:56:38 +0300 |
| commit | 3fa084f90a9f04adf70e240a85a0c88c9ad2b58e (patch) | |
| tree | 0c2c15c6325dea6aef56a884f665a7b0a4ed1452 /scripts | |
| parent | 4ba7202ae557490d1b8b1116bddfcf67f7a91017 (diff) | |
Fix corpus in word2vec script
Diffstat (limited to 'scripts')
| -rwxr-xr-x | scripts/embeddings/train_word2vec.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/scripts/embeddings/train_word2vec.py b/scripts/embeddings/train_word2vec.py index d827446e..c34c0d2a 100755 --- a/scripts/embeddings/train_word2vec.py +++ b/scripts/embeddings/train_word2vec.py @@ -46,7 +46,7 @@ def main(): orig_vectors = args.output + '.orig' cmd = "{w2v} {opts} -train {i} -output {o} -size {s} -threads {t}" \ .format(w2v=args.word2vec, opts=WORD2VEC_OPTIONS, - i=args.corpus, o=orig_vectors, s=args.dim_emb, t=args.threads) + i=prep_corpus, o=orig_vectors, s=args.dim_emb, t=args.threads) print(" with command: {}".format(cmd)) proc = subprocess.Popen(cmd, shell=True) |