Fix vocabulary naming (input/output vocabulary no longer has same name) (#946)

efa4a6cf · Justas Zemgulys · drpngx · 5723fcca · efa4a6cf
Commit efa4a6cf authored Jan 24, 2017 by Justas Zemgulys Committed by drpngx Jan 24, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tutorials/rnn/translate/data_utils.py tutorials/rnn/translate/data_utils.py +2 -2

No files found.
--- a/tutorials/rnn/translate/data_utils.py
+++ b/tutorials/rnn/translate/data_utils.py
@@ -300,8 +300,8 @@ def prepare_data(data_dir, from_train_path, to_train_path, from_dev_path, to_dev
        (6) path to the "to language" vocabulary file.
    """
  # Create vocabularies of the appropriate sizes.
-  to_vocab_path = os.path.join(data_dir, "vocab%d" % to_vocabulary_size)
-  from_vocab_path = os.path.join(data_dir, "vocab%d" % from_vocabulary_size)
+  to_vocab_path = os.path.join(data_dir, "vocab%d.to" % to_vocabulary_size)
+  from_vocab_path = os.path.join(data_dir, "vocab%d.from" % from_vocabulary_size)
  create_vocabulary(to_vocab_path, to_train_path , to_vocabulary_size, tokenizer)
  create_vocabulary(from_vocab_path, from_train_path , from_vocabulary_size, tokenizer)