开始使用免费开始使用

训练 GB 回归器

现在,您将训练在上一个练习中实例化的梯度提升回归器 gb,并对测试集的标签进行预测。

数据集已按 80% 训练集和 20% 测试集划分。特征矩阵 X_trainX_test,以及数组 y_trainy_test 已在您的工作区中可用。此外,我们还为您加载了上一个练习中定义的模型实例 gb

本练习是课程的一部分

Python 树模型机器学习

查看课程

练习说明

  • gb 拟合到训练集。
  • 预测测试集标签,并将结果赋给 y_pred

交互式实操练习

通过完成这段示例代码来试试这个练习。

# Fit gb to the training set
____

# Predict test set labels
y_pred = ____
编辑并运行代码