Prevents TypeError:'dict_keys' object does not support indexing

eb784c09 · Brendan Apfeld · e2fb51d1 · eb784c09
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

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

未找到文件。
--- 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()