开始使用免费开始使用

您的第一个 AdaBoost 模型

在上一课中,您构建了用于预测电影对数收入的模型。您先从简单的线性回归开始,得到的 RMSE 为 7.34。随后,您尝试进行一次提升(boosting)迭代,将 RMSE 降至 7.28

在本练习中,您将构建第一个 AdaBoost 模型——AdaBoostRegressor,以期进一步提升性能。

movies 数据集已加载并拆分为训练集和测试集。这里您将使用 'budget''popularity' 两个特征,这些特征已使用 sklearn.preprocessing 模块中的 StandardScaler() 为您标准化。

本练习是课程的一部分

Python 中的集成方法

查看课程

练习说明

  • 实例化默认的线性回归模型。
  • 构建并拟合一个 AdaBoostRegressor,将线性回归作为基学习器,并使用 12 个估计器。
  • 在测试集上计算预测结果。

交互式实操练习

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

# Instantiate the default linear regression model
reg_lm = ____

# Build and fit an AdaBoost regressor
reg_ada = ____(____, ____, random_state=500)
reg_ada.fit(X_train, y_train)

# Calculate the predictions on the test set
pred = ____

# Evaluate the performance using the RMSE
rmse = np.sqrt(mean_squared_error(y_test, pred))
print('RMSE: {:.3f}'.format(rmse))
编辑并运行代码