Fix get_word_index (keras-team#2981)

lemonhall · Jun 15, 2016 · c53c64d · c53c64d
1 parent 6b122ba
commit c53c64d
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/keras/datasets/reuters.py b/keras/datasets/reuters.py
@@ -4,6 +4,7 @@
 from six.moves import cPickle
 from six.moves import zip
 import numpy as np
+import sys
 
 
 def load_data(path="reuters.pkl", nb_words=None, skip_top=0,
@@ -64,4 +65,11 @@ def load_data(path="reuters.pkl", nb_words=None, skip_top=0,
 def get_word_index(path="reuters_word_index.pkl"):
     path = get_file(path, origin="https://s3.amazonaws.com/text-datasets/reuters_word_index.pkl")
     f = open(path, 'rb')
-    return cPickle.load(f)
+
+    if sys.version_info < (3,):
+        data = cPickle.load(f)
+    else:
+        data = cPickle.load(f, encoding="latin1")
+
+    f.close()
+    return data