Cómo extraer los mejores parámetros de un CrossValidatorModel
Quiero encontrar los parámetros deParamGridBuilder
que hacen el mejor modelo en CrossValidator en Spark 1.4.x,
EnEjemplo de tubería en la documentación de Spark, agregan diferentes parámetros (numFeatures
, regParam
) mediante el usoParamGridBuilder
En la tuberia. Luego, por la siguiente línea de código, hacen el mejor modelo:
val cvModel = crossval.fit(training.toDF)
Ahora, quiero saber cuáles son los parámetros (numFeatures
, regParam
) deParamGridBuilder
que produce el mejor modelo.
Ya usé los siguientes comandos sin éxito:
cvModel.bestModel.extractParamMap().toString()
cvModel.params.toList.mkString("(", ",", ")")
cvModel.estimatorParamMaps.toString()
cvModel.explainParams()
cvModel.getEstimatorParamMaps.mkString("(", ",", ")")
cvModel.toString()
¿Alguna ayuda?
Gracias por adelantado,