Advertisement

机器学习项目——音乐推荐系统代码

阅读量:

内容源自coursera
相关的代码及数据资料可通过这里下载获取。

复制代码
    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)

还没有任何评论哟~