開始使用免費開始

整體驗證分數

現在是用交叉驗證來取得實際模型效能的時候了!我們的門市商品需求預測模型表現如何?

你的任務是分別取得每個摺疊(fold)的均方誤差(MSE),然後把這些結果彙整成單一數值。

為了簡化,你已經拿到 get_fold_mse() 函式。它會在每個交叉驗證切分上訓練一個 Random Forest 模型,並回傳依照摺疊列出的 MSE 分數清單。get_fold_mse() 接受 2 個引數:trainTimeSeriesSplit 物件。

本練習屬於課程

用 Python 拿下 Kaggle 競賽

檢視課程

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

from sklearn.model_selection import TimeSeriesSplit
import numpy as np

# Sort train data by date
train = train.sort_values('date')

# Initialize 3-fold time cross-validation
kf = ____(n_splits=____)

# Get MSE scores for each cross-validation split
mse_scores = get_fold_mse(train, kf)

print('Mean validation MSE: {:.5f}'.format(np.____(____)))
編輯並執行程式碼