Prevents TypeError:'dict_keys' object does not support indexing

eb784c09 · Brendan Apfeld · e2fb51d1 · eb784c09
Commit eb784c09 authored Oct 11, 2018 by Brendan Apfeld
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

research/skip_thoughts/skip_thoughts/data/preprocess_dataset.py ...ch/skip_thoughts/skip_thoughts/data/preprocess_dataset.py +2 -2

No files found.
--- a/research/skip_thoughts/skip_thoughts/data/preprocess_dataset.py
+++ b/research/skip_thoughts/skip_thoughts/data/preprocess_dataset.py
@@ -121,8 +121,8 @@ def _build_vocabulary(input_files):

  tf.logging.info("Processed %d sentences total", num)

-  words = wordcount.keys()
-  freqs = wordcount.values()
+  words = list(wordcount)
+  freqs = list(wordcount.values())
  sorted_indices = np.argsort(freqs)[::-1]

  vocab = collections.OrderedDict()