java .lang.Integer не может быть приведен к java .lang. Двойная логистика c регрессия

Question

У меня есть логистика c регрессионная модель в искре и вычисление RMSE на ней

val fullPredictions = model.transform(test).cache()
    val prediction = fullPredictions.select("prediction").rdd.map(_.getDouble(0))
    val labels = fullPredictions.select("label").rdd.map(_.getDouble(0))
    val RMSE = new RegressionMetrics(prediction.zip(labels)).rootMeanSquaredError
  println(s"  Root mean squared error (RMSE): $RMSE")

это исключение появляется

java .lang.ClassCastException: java .lang.Integer не может быть приведен к java .lang.Double

Как решить?