开始使用免费开始使用

训练一棵决策树

现在,请尝试构建一个决策树模型。决策树是一系列通过机器学习得到的 if-else 规则。在电信客户流失的案例中,它用于判断客户是否会流失。下面是一个基于著名 Titanic 幸存数据集构建的决策树示例图。

上一练习中的 train_Xtest_Xtrain_Ytest_Y 已为您加载。同时,已从 sklearn 库中加载了 tree 模块和 accuracy_score 函数。接下来,您将构建模型,并在未见过的数据上检验其性能。

本练习是课程的一部分

Python 营销中的机器学习

查看课程

练习说明

  • 使用 max_depth 设为 5 来初始化决策树模型。
  • 在训练数据上拟合模型,顺序为先 train_X,后 train_Y
  • 对测试数据进行预测,这里为 test_X
  • 通过比较实际测试标签与预测标签来评估模型在测试数据上的性能。

交互式实操练习

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

# Initialize the model with max_depth set at 5
mytree = tree.___(max_depth = ___)

# Fit the model on the training data
treemodel = mytree.___(___, ___)

# Predict values on the testing data
pred_Y = treemodel.___(___)

# Measure model performance on testing data
accuracy_score(___, ___)
编辑并运行代码