开始使用免费开始使用

获取预测并进行首次评估

现在我们已经训练好了随机森林模型(rfr),接下来要用它在测试集上进行预测。这样做是为了评估模型表现——基本判断是:它是否不差于直接买入指数 SPY,或者更好?

我们将使用常见的 sklearn .predict(features) 方法,然后把月度收益与投资组合预测值相乘。由于每个月会有 3 行,我们用 np.sum() 进行求和。最后绘制基于我们预测得到的月度收益曲线,并与 SPY 的月度收益一起对比。

本练习是课程的一部分

Python 金融机器学习

查看课程

练习说明

  • 使用随机森林模型 rfr.predict() 方法,分别对 train_featurestest_features 进行预测。
  • returns_monthly 的测试集部分与 test_predictions 相乘,得到测试集预测的收益。
  • 绘制测试集期间 'SPY'returns_monthly(从 train_size 到数据末尾)。

交互式实操练习

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

# Get predictions from model on train and test
train_predictions = rfr.predict(train_features)
test_predictions = ____

# Calculate and plot returns from our RF predictions and the SPY returns
test_returns = np.sum(returns_monthly.iloc[train_size:] * ____, axis=1)
plt.plot(test_returns, label='algo')
plt.plot(returns_monthly['SPY'].iloc[____], label='SPY')
plt.legend()
plt.show()
编辑并运行代码