Fixed default dataset value in load_data()

carriepl · carriepl · commit 04c02d4f889a · 2015-01-13T11:27:43.000-05:00
diff --git a/code/imdb.py b/code/imdb.py
@@ -87,7 +87,7 @@ def load_data(path="imdb.pkl", n_words=100000, valid_portion=0.1):
 
     # Load the dataset
     path = get_dataset_file(
-        path, "imdb.pkl.gz",
+        path, "imdb.pkl",
         "http://www.iro.umontreal.ca/~lisa/deep/data/imdb.pkl")
 
     if path.endswith(".gz"):
