Scala spark中svm的权值初始化

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

我正在尝试为spark中的SVM初始化分配自己的权重。但是我在训练模型后得到了错误的预测值。如果我使用默认值,即不提供初始权重向量,我会得到一个好的预测模型

我在github上检查了代码,但无法弄清楚在我们不传递任何消息的情况下如何进行初始化。

有什么建议吗


编辑:我对scala有0点经验

SVMWithSGD
从一个。如果未提供权重向量,则其仅为:

val initialWeights = {
  if (numOfLinearPredictor == 1) {
    Vectors.zeros(numFeatures)
  } else if (addIntercept) {
    Vectors.zeros((numFeatures + 1) * numOfLinearPredictor)
  } else {
    Vectors.zeros(numFeatures * numOfLinearPredictor)
  }
}