############################################################################### # load origin data and label data_path = '/Users/cengqiqi/Desktop/project/data/' track_features = pd.read_csv(data_path + 'tf_mini.csv') sessions = pd.read_csv(data_path + 'log_mini.csv') # get track dict track_dic = get_track_dic(track_features) ntracks = len(track_dic) # get train data, validation data and test data # Note: you should build track dictionary using function get_track_dic before get data data_raw, label, sf_pos = get_data(batch_size, sessions, seq_len) train_data_raw = data_raw[0:15000] val_data_raw = data_raw[15000:16000] test_data_raw = data_raw[16000:16660] train_label = label[0:15000] val_label = label[15000:16000] test_label = label[16000:16660] train_sf = sf_pos[0:15000] val_sf = sf_pos[15000:16000] test_sf = sf_pos[16000:16660] ############################################################################### # prepare data for track embedding in pytorch
############################################################################### data_path = '/Users/cengqiqi/Desktop/project/data/' # load origin data and label track_features = pd.read_csv(data_path + 'tf_mini.csv') sessions = pd.read_csv(data_path + 'log_mini.csv') # get track dict track_dic = get_track_dic(track_features) ntracks = len(track_dic) # get train data, validation data and test data # Note: you should build track dictionary using function get_track_dic before get data data_raw, label, sf_pos = get_data(args.batch_size, sessions, args.seq_len) train_data_raw = data_raw[0:15000] val_data_raw = data_raw[15000:16000] test_data_raw = data_raw[16000:16660] train_label = label[0:15000] val_label = label[15000:16000] test_label = label[16000:16660] train_sf = sf_pos[0:15000] val_sf = sf_pos[15000:16000] test_sf = sf_pos[16000:16660] ############################################################################### # prepare data for track embedding in pytorch