训练 GB 回归器
现在,您将训练在上一个练习中实例化的梯度提升回归器 gb,并对测试集的标签进行预测。
数据集已按 80% 训练集和 20% 测试集划分。特征矩阵 X_train 与 X_test,以及数组 y_train 和 y_test 已在您的工作区中可用。此外,我们还为您加载了上一个练习中定义的模型实例 gb。
本练习是课程的一部分
Python 树模型机器学习
练习说明
- 将
gb拟合到训练集。 - 预测测试集标签,并将结果赋给
y_pred。
交互式实操练习
通过完成这段示例代码来试试这个练习。
# Fit gb to the training set
____
# Predict test set labels
y_pred = ____