From 5ea86769b33c2dc1687e5a3d3508126e65fedd8b Mon Sep 17 00:00:00 2001 From: Chong Chen Date: Sat, 14 Apr 2018 22:09:04 +0800 Subject: [PATCH] Update loaddata.py --- pro_data/loaddata.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pro_data/loaddata.py b/pro_data/loaddata.py index 81e514d..f4fe723 100644 --- 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={}