1. Learn
  2. /
  3. 课程
  4. /
  5. Python 树模型机器学习

Connected

道练习

评估 10 折交叉验证误差

在本练习中,您将评估上一题中实例化的回归树 dt 所取得的 10 折交叉验证均方根误差(RMSE)。

除 dt 外,训练数据 X_train 和 y_train 也已在您的工作区中可用。我们还从 sklearn.model_selection 导入了 cross_val_score。

请注意,由于 cross_val_score 只能返回负的 MSE,其输出需要乘以 -1 才能得到 MSE。随后对这些 MSE 取平均并开平方即可得到交叉验证 RMSE。

说明

100 XP
  • 通过将 scoring 参数设为 'neg_mean_squared_error',计算 dt 的 10 折交叉验证 MSE。

  • 根据得到的 MSE 分数计算 RMSE。