开始使用免费开始使用

训练您的第一个回归树

在本练习中,您将训练一棵回归树,用所有 6 个可用特征来预测 auto-mpg 数据集中汽车的 mpg(每加仑英里数)。

数据集已为您预处理,并按 80% 训练集和 20% 测试集划分。特征矩阵 X_train 和数组 y_train 已在您的工作区中提供。

本练习是课程的一部分

Python 树模型机器学习

查看课程

练习说明

  • sklearn.tree 导入 DecisionTreeRegressor
  • 实例化一个 DecisionTreeRegressor dt,最大深度设为 8,min_samples_leaf 设为 0.13。
  • dt 拟合到训练集。

交互式实操练习

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

# Import DecisionTreeRegressor from sklearn.tree
from ____.____ import ____

# Instantiate dt
____ = ____(max_depth=____,
             ____=____,
            random_state=3)

# Fit dt to the training set
____.____(____, ____)
编辑并运行代码