开始使用免费开始使用

拟合一棵决策树

随机森林是常用的预测模型,开箱即用且效果不错。不过我们先来学习随机森林的基础模块——决策树。

决策树会根据特征将数据划分为不同的组。决策树从根节点开始,一直向下划分,直到到达叶节点。

decision tree

我们可以使用 sklearn 中的 DecisionTreeRegressor.fit(features, targets) 来拟合一棵决策树。

如果不限制树的深度(或高度),它会不断划分数据,直到每个叶节点只有 1 个样本,这就是典型的过拟合。我们将在后续章节进一步学习过拟合。

本练习是课程的一部分

Python 金融机器学习

查看课程

练习说明

  • 使用已导入的 DecisionTreeRegressor 类,采用默认参数(即不传参)创建一个名为 decision_tree 的决策树模型。
  • 使用我们之前创建的 train_featurestrain_targets(其中已包含星期和成交量特征)来拟合该模型。
  • 分别打印模型在训练集特征与目标,以及在 test_featurestest_targets 上的得分。

交互式实操练习

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

from sklearn.tree import DecisionTreeRegressor

# Create a decision tree regression model with default arguments
decision_tree = ____

# Fit the model to the training features and targets
decision_tree.fit(____)

# Check the score on train and test
print(decision_tree.score(train_features, train_targets))
print(decision_tree.score(____))
编辑并运行代码