加载crossvalidator对象pyspark

yfjy0ee7  于 2021-05-27  发布在  Spark
关注(0)|答案(1)|浏览(404)

我得到一个错误,说cvmodel.bestmodel没有培训摘要。我可以访问CVU套索的摘要,但一旦我保存并加载它,我就无法查看摘要。如何查看此摘要?

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
9rnv2umw

9rnv2umw1#

通常,您会保存最好的模型,而不是像您所做的那样保存验证器模型。

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

在加载回 best_model ,您应该能够检索摘要。

相关问题