Como extrair os melhores parâmetros de um CrossValidatorModel
Eu quero encontrar os parâmetros deParamGridBuilder
que fazem o melhor modelo no CrossValidator no Spark 1.4.x,
NoExemplo de Pipeline na documentação do Spark, eles adicionam parâmetros diferentes (numFeatures
, regParam
) usandoParamGridBuilder
no pipeline. Então, pela seguinte linha de código, eles formam o melhor modelo:
val cvModel = crossval.fit(training.toDF)
Agora, quero saber quais são os parâmetros (numFeatures
, regParam
) deParamGridBuilder
que produz o melhor modelo.
Eu já usei os seguintes comandos sem sucesso:
cvModel.bestModel.extractParamMap().toString()
cvModel.params.toList.mkString("(", ",", ")")
cvModel.estimatorParamMaps.toString()
cvModel.explainParams()
cvModel.getEstimatorParamMaps.mkString("(", ",", ")")
cvModel.toString()
Qualquer ajuda?
Desde já, obrigado,