加载CrossValidator对象Pyspark

Question

我得到一个错误，说没有cvModel.bestModel的训练摘要。我可以访问cvModel_lasso的摘要，但是一旦保存并加载它，就无法查看摘要。我如何查看此摘要？

from pyspark.ml.tuning import ParamGridBuilder, CrossValidator, CrossValidatorModel

cv = CrossValidator(estimator=lr, estimatorParamMaps=paramGrid, evaluator=regEval, numFolds=10)

cvModel_lasso = cv.fit(preppedDataDF)

cvModel_lasso.write().overwrite().save('myModelPath_lasso')
cvModel=CrossValidatorModel.load('myModelPath_lasso')

trainingSummary = cvModel.bestModel.summary

收到的错误：

RuntimeError: No training summary available for this LinearRegressionModel

Answer 1

通常，您将保存最佳模型，而不是像完成时那样保存验证器模型。

best_model = cvModel_lasso.bestModel
best_model.write().overwrite().save('myModelPath_lasso')

在加载best_model时，您应该能够检索摘要。