Graphlab knn协同过滤的错误预测

时间:2018-04-18 03:40:26

标签: graphlab recommender-systems

我正在使用Graphlab中的item_similarity_recommender,我发现即使收视率从1-5开始,predict()函数也会为大多数预测项目指定零。这当然会导致RMSE非常高。

由于item_similarity_recommender基于KNN,我认为预测的评级应该是1-5。任何人都可以向我解释为什么会发生这种情况吗?

代码如下:

train, test = graphlab.recommender.util.random_split_by_user(dataset, max_num_users=1000)
m = graphlab.recommender.item_similarity_recommender.create(train, target='target', only_top_k=65)
rmse=m.evaluate_rmse(test, target='target')
print(m.predict(test))

0 个答案:

没有答案