开始使用免费开始使用

保存与加载模型

很多时候,您需要回到之前训练过的模型,查看当时用于诊断预测误差来源的假设或设置。也许是数据有问题?或者需要加入一个新特征来捕捉某次异常事件?

在本练习中,您将练习如何保存和加载模型。

本练习是课程的一部分

使用 PySpark 进行特征工程

查看课程

练习说明

  • pyspark.ml.regression 导入 RandomForestRegressionModel
  • 使用内存中的模型 model,调用其 save() 方法,并将模型命名为 rfr_no_listprice
  • 通过在 RandomForestRegressionModel 上调用 load() 重新加载已保存的模型文件 rfr_no_listprice,并将其保存到 loaded_model 中。

交互式实操练习

通过完成这段示例代码来试试这个练习。

from ____ import ____

# Save model
model.____(____)

# Load model
loaded_model = ____.____(____)
编辑并运行代码