机器学习项目——音乐推荐系统代码
发布时间
阅读量:
阅读量
import graphlab
song_data = graphlab.SFrame('song_data.gl/')
graphlab.canvas.set_target('ipynb')
#查看有多少用户
users = song_data['user_id'].unique()
#数据分为训练集和测试集
train_data,test_data = song_data.random_split(0.8,seed=0)
#建立一个基于受欢迎程度的模型(无法克服流行歌曲推荐力过强)
popularity_model = graphlab.popularity_recommender.create(train_data,user_id='user_id',item_id='song')
#给特定用户推荐歌曲
popularity_model.recommend(users=[users[0]])
popularity_model.recommend(us
全部评论 (0)
还没有任何评论哟~
