Fix README for Netflix Test

.gitignore

@@ -1,3 +1,9 @@
+# Data files and folders
+download/
+training_set/
+nf_prize_dataset.tar.gz
+Netflix/
+
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]

README.md

@@ -22,29 +22,16 @@ $ python -m unittest test/test_model.py
 
 ### Get the data
 
+**Note: Run all these commands within your `DeepRecommender` folder**
+
 [Netflix prize](http://netflixprize.com/)
 
-* ```$ mkdir -p ~/Recommendations``` you can use any other folder name
-* Download from [here](http://academictorrents.com/details/9b13183dc4d60676b773c9e2cd6de5e5542cee9a) to ```~/Recommendations```
-* ```$ cd ~/Recommendations```
-* ```$ tar -xvf nf_prize_dataset.tar.gz```
-* ```$ tar -xf download/training_set.tar ```
-* Create necessary folders
+* Download from [here](http://academictorrents.com/details/9b13183dc4d60676b773c9e2cd6de5e5542cee9a) into your ```DeepRecommender``` folder
 ```
-mkdir -p Netflix/N3M_TRAIN
-mkdir -p Netflix/N3M_VALID
-mkdir -p Netflix/N3M_TEST
-mkdir -p Netflix/N6M_TRAIN
-mkdir -p Netflix/N6M_VALID
-mkdir -p Netflix/N6M_TEST
-mkdir -p Netflix/N1Y_TRAIN
-mkdir -p Netflix/N1Y_VALID
-mkdir -p Netflix/N1Y_TEST
-mkdir -p Netflix/NF_TRAIN
-mkdir -p Netflix/NF_VALID
-mkdir -p Netflix/NF_TEST
+$ tar -xvf nf_prize_dataset.tar.gz
+$ tar -xf download/training_set.tar
+$ python ./data_utils/netflix_data_convert.py training_set Netflix
 ```
-* ```$ python ./data_utils/netflix_data_convert.py training_set Netflix```. (run from your `DeepRecommender` folder)
 
 #### Data stats
 | Dataset  | Netflix 3 months | Netflix 6 months | Netflix 1 year | Netflix full |

data_utils/netflix_data_convert.py

@@ -1,5 +1,5 @@
 # Copyright (c) 2017 NVIDIA Corporation
-from os import listdir, path
+from os import listdir, path, makedirs
 import random
 import sys
 import time
@@ -75,6 +75,17 @@ def create_NETFLIX_data_timesplit(all_data,
 
 
 def main(args):
+  # create necessary folders:
+  for output_dir in [
+    "Netflix/N3M_TRAIN", "Netflix/N3M_VALID", "Netflix/N3M_TEST", "Netflix/N6M_TRAIN",
+    "Netflix/N6M_VALID", "Netflix/N6M_TEST", "Netflix/N1Y_TRAIN", "Netflix/N1Y_VALID",
+    "Netflix/N1Y_TEST", "Netflix/NF_TRAIN", "Netflix/NF_VALID", "Netflix/NF_TEST"]:
+    try:
+       makedirs(output_dir)
+    except OSError as exception:
+      if exception.errno != errno.EEXIST:
+        raise
+
   user2id_map = dict()
   item2id_map = dict()
   userId = 0