开始使用免费开始使用

整体验证分数

现在该通过交叉验证来获得模型的实际表现了!我们的门店商品需求预测模型表现如何?

您的任务是分别计算每个折的均方误差(MSE),然后把这些结果合并为一个数字。

为简化操作,已为您提供 get_fold_mse() 函数。它会在每个交叉验证切分上拟合一个随机森林模型,并按折返回 MSE 分数组成的列表。get_fold_mse() 接受两个参数: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.____(____)))
编辑并运行代码