Fix the dataset loading when in the code directory.

Author: nouiz

code/logistic_sgd.py

@@ -159,8 +159,12 @@ def load_data(dataset):
     data_dir, data_file = os.path.split(dataset)
     if data_dir == "" and not os.path.isfile(dataset):
         # Check if dataset is in the data directory.
-        new_path = os.path.join(os.path.split(os.path.split(__file__)[0])[0],
-                                "data", dataset)
+        new_path = os.path.split(__file__)
+        if new_path[0] == "":
+            new_path = os.path.join('..', "data", dataset)
+        else:
+            new_path = os.path.join(os.path.split(os.path.split(__file__)[0])[0],
+                                    "data", dataset)
         if os.path.isfile(new_path) or data_file == 'mnist.pkl.gz':
             dataset = new_path