Scala spark中svm的权值初始化
我正在尝试为spark中的SVM初始化分配自己的权重。但是我在训练模型后得到了错误的预测值。如果我使用默认值,即不提供初始权重向量,我会得到一个好的预测模型 我在github上检查了代码,但无法弄清楚在我们不传递任何消息的情况下如何进行初始化。 有什么建议吗Scala spark中svm的权值初始化,scala,apache-spark,Scala,Apache Spark,我正在尝试为spark中的SVM初始化分配自己的权重。但是我在训练模型后得到了错误的预测值。如果我使用默认值,即不提供初始权重向量,我会得到一个好的预测模型 我在github上检查了代码,但无法弄清楚在我们不传递任何消息的情况下如何进行初始化。 有什么建议吗 编辑:我对scala有0点经验SVMWithSGD从一个。如果未提供权重向量,则其仅为: val initialWeights = { if (numOfLinearPredictor == 1) { Vectors.zero
编辑:我对scala有0点经验
SVMWithSGD
从一个。如果未提供权重向量,则其仅为:
val initialWeights = {
if (numOfLinearPredictor == 1) {
Vectors.zeros(numFeatures)
} else if (addIntercept) {
Vectors.zeros((numFeatures + 1) * numOfLinearPredictor)
} else {
Vectors.zeros(numFeatures * numOfLinearPredictor)
}
}