update the data split function for song recommendation