1. Learn
    2. /
  2. 课程
    3. /
  3. Python 树模型机器学习
Connected

道练习

训练您的第一个回归树

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

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

说明

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