【发布时间】:2015-03-24 14:39:58
【问题描述】:
尝试运行我的方法时:
def doGD() = {
allRatings.foreach(rating => gradientDescent(rating));
}
我收到错误:org.apache.spark.SparkException: Task not serialisable
我知道我的梯度下降方法不会并行化,因为每个步骤都取决于上一步 - 所以并行工作不是一种选择。但是,如果我这样做,从控制台:
val gd = new GradientDescent()
gd.doGD();
我收到上述错误。
但是,如果在控制台中我这样做:
val gd = new GradientDescent()
gd.allRatings.foreach(rating => gradientDescent(rating))
它工作得很好。您可能已经注意到,第二个示例中的代码与第一个示例中的代码相同,只是我只是将代码从方法中取出并直接调用它而不是方法。
为什么一个有效而另一个无效?我很困惑。
(补充说明:GradientDescent extends Serializable 类)。
gradientDescent 方法:
def gradientDescent(rating : Rating) = {
var userVector = userFactors.get(rating.user).get
var itemVector = itemFactors.get(rating.product).get
userFactors.map(x => if(x._1 == rating.user)(x._1, x._2 += 0.02 * (calculatePredictionError(rating.rating, userVector, itemVector) * itemVector)))
userVector = userFactors.get(rating.user).get // updated user vector
itemFactors.map(x => if(x._1 == rating.product)(x._1, x._2 += 0.02 * (calculatePredictionError(rating.rating, userVector, itemVector) * itemVector)))
}
我知道我正在使用存储在主服务器上的 2 个变量 - userFactors 和 itemFactors - 由于该过程是顺序并行化的,因此不可能。但这并不能解释为什么从控制台调用方法不起作用但在控制台中重写方法的内部。
【问题讨论】:
标签: apache-spark rdd gradient-descent