【发布时间】:2017-03-16 07:45:58
【问题描述】:
我有一个名为 train.dat 的文件,其中包含三个字段 - userID、movieID 和 rating。
我需要根据这个预测test.dat文件中的rating。
鉴于我只有一个功能 - rating,我想知道如何使用 scikit-learn 的 KMeans 对类似用户进行分组。
这甚至有意义吗?在聚类步骤之后,我可以执行回归步骤来获取 test.dat 中每个用户电影对的评分
编辑:我有一些额外的文件,其中包含每部电影中的演员、导演以及电影所属的类型。我不确定如何开始使用这些,我问这个问题是因为我想知道是否有可能获得一个仅使用评级的简单模型,然后使用其他数据对其进行增强。我读到这被称为基于内容的推荐。对不起,我应该也写一下其他数据文件。
【问题讨论】:
标签: python machine-learning scikit-learn k-means