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

道练习

评估 RF 回归器

现在，您将评估上一练习中训练的随机森林回归器 rf 在测试集上的 RMSE。

数据集已为您预处理，并按 80% 训练集、20% 测试集进行拆分。特征矩阵 X_test 和数组 y_test 已在您的工作区中可用。另外，我们也已加载了您在上一练习中训练的模型 rf。

说明

100 XP
  • 从 sklearn.metrics 中以 MSE 的别名导入 mean_squared_error。
  • 预测测试集标签，并将结果赋给 y_pred。
  • 计算测试集的 RMSE，并将其赋给 rmse_test。