开始使用免费开始使用

拟合随机森林

数据科学家常用随机森林模型。它开箱即用表现良好,并且有很多可调参数来优化性能。随机森林既可用于分类,也可用于回归;我们将在这里用于回归,以预测 LNG 的未来价格变动。

我们将像使用决策树一样创建并拟合随机森林模型,使用 .fit(features, targets) 方法。借助 sklearn 的 RandomForestRegressor,可以使用内置的 .score() 方法评估性能。该方法接收参数(featurestargets),返回 R\(^2\) 分数(判定系数)。

本练习是课程的一部分

Python 金融机器学习

查看课程

练习说明

  • 使用已导入的 RandomForestRegressor 类创建随机森林模型。
  • 使用 train_featurestrain_targets 拟合(训练)随机森林。
  • 打印训练集和测试集的 R\(^2\) 分数。

交互式实操练习

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

from sklearn.ensemble import RandomForestRegressor

# Create the random forest model and fit to the training data
rfr = ____(n_estimators=200)
rfr.fit(____, ____)

# Look at the R^2 scores on train and test
print(rfr.score(train_features, train_targets))
print(____)
编辑并运行代码