加载 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 时，您应该能够检索摘要。