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={}