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

道练习

评估 GB 回归器

现在已经得到了测试集的预测值，您可以用它们来评估模型 gb 在测试集上的均方根误差（RMSE）。

y_test 和预测值 y_pred 已在您的工作区中提供。

说明

100 XP

  • 从 sklearn.metrics 导入 mean_squared_error，并命名为 MSE。

  • 计算测试集的 MSE，赋值给 mse_test。

  • 计算测试集的 RMSE，赋值给 rmse_test。