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