モデルの保存と読み込み
予測誤差の原因を調べるとき、以前のモデルに戻って、そのときの前提や設定を確認したくなることがよくあります。もしかするとデータに問題があったのかもしれませんし、発生した特異な事象を捉えるために新しい特徴量を加える必要があるのかもしれません。
この演習では、モデルの保存と読み込みを練習します。
この演習はコースの一部です
PySparkで学ぶ特徴量エンジニアリング
演習の手順
pyspark.ml.regressionからRandomForestRegressionModelをインポートします。- メモリ上のモデル
modelに対してsave()メソッドを呼び出し、モデル名をrfr_no_listpriceとして保存します。 - 保存したモデルファイル
rfr_no_listpriceを、RandomForestRegressionModelのload()を呼び出して読み込み、loaded_modelに代入します。
実践的なインタラクティブ演習
このサンプルコードを完成させて、この演習に挑戦してみましょう。
from ____ import ____
# Save model
model.____(____)
# Load model
loaded_model = ____.____(____)