Spark ALS Python-设置预测范围

时间:2019-03-13 19:10:06

标签: python apache-spark pyspark

我正在研究老师的建议,以便使用Spark编写的ALS ALS创建电影评级预测器。我接近令人满意的结果,但检查预测输出后,我发现,当作为参数传递给模型的评分介于1到5之间时,某些预测的值超过5。

因此,我想知道自己是否可以在设置ALS函数时为预测值固定范围,因为我可以设置的唯一预测值是预测为正值。

这实际上是我的ALS函数的参数,我已经进行了一些搜索,但是没有找到与实现我想要的目标相关的任何东西。

ALS(userCol="User ID", itemCol="Film ID", ratingCol="label", coldStartStrategy="drop", seed=1234, nonnegative=True)

谢谢。

0 个答案:

没有答案