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

exercițiu

线性回归 vs 回归树

在本练习中，您将比较回归树 dt 在测试集上的 RMSE 与线性回归模型的表现。我们已经实例化了线性回归模型 lr，并用与 dt 相同的数据集进行了训练。

工作区中已提供特征矩阵 X_test、标签数组 y_test、已训练好的线性回归模型 lr、以别名 MSE 导入的 mean_squared_error 函数，以及上一个练习得到的 rmse_dt。

Instrucțiuni

100 XP

  • 使用线性回归模型（lr）预测测试集标签，并将结果赋给 y_pred_lr。

  • 计算测试集的 MSE，并将结果赋给 mse_lr。

  • 计算测试集的 RMSE，并将结果赋给 rmse_lr。