Update loaddata.py

jeongin601 · Apr 14, 2018 · 5ea8676 · 5ea8676
1 parent 22aca0a
commit 5ea8676
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pro_data/loaddata.py b/pro_data/loaddata.py
@@ -80,9 +80,9 @@ def numerize(tp):
 
 tp_test = tp_1[test_idx]
 tp_valid = tp_1[~test_idx]
-tp_train.to_csv(os.path.join(TPS_DIR, 'Toys_train.csv'), index=False,header=None)
-tp_valid.to_csv(os.path.join(TPS_DIR, 'Toys_valid.csv'), index=False,header=None)
-tp_test.to_csv(os.path.join(TPS_DIR, 'Toys_test.csv'), index=False,header=None)
+tp_train.to_csv(os.path.join(TPS_DIR, 'music_train.csv'), index=False,header=None)
+tp_valid.to_csv(os.path.join(TPS_DIR, 'music_valid.csv'), index=False,header=None)
+tp_test.to_csv(os.path.join(TPS_DIR, 'music_test.csv'), index=False,header=None)
 
 user_reviews={}
 item_reviews={}