获取TrainValidationSplit scala的最佳参数

时间:2017-05-09 23:32:10

标签: scala apache-spark apache-spark-mllib apache-spark-ml

我正在使用Spark Scala ML API,我正在尝试将管道ALS模型传递给TrainValidationSplit。代码执行但我无法检索最佳参数...想法?

  e1.t * required_width + e1.width < e2.t * required_width

1 个答案:

答案 0 :(得分:1)

您可以使用以下网址获取网格中每个参数的rmse

alsModel.getEstimatorParamMaps.zip(alsModel.avgMetrics)