def generate_interaction(_tr, _tv, _tp):
print "Creating user song-interaction lists"
_, all_songs = MSD_util.get_unique(_tr, users=False, songs=True)
testv_pairs, testp_pairs = MSD_util.get_user_song_pairs(_tv, _tp)
return all_songs, testv_pairs, testp_pairs
def generate_interaction(_tr, _va):
print "Creating user song-interaction lists"
_, all_songs = MSD_util.get_unique(_tr, users=False, songs=True)
train_pairs, valid_pairs = MSD_util.get_user_song_pairs(_tr, _va)
return all_songs, train_pairs, valid_pairs
