开始使用免费开始使用

与单一分类器对比

您已经了解到,提升树是目前最强的机器学习算法之一。为此,您现在将把提升模型的 AUC 与一个简单决策树分类器进行对比。

在本练习中,您将使用完整的机器学习流程,从模型设定到评估。

训练数据集 customers_train 已经预加载。

本练习是课程的一部分

R 中的树模型机器学习

查看课程

交互式实操练习

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

set.seed(100)

# Specify, fit, predict, and combine with training data
predictions <- boost_tree() %>%
  set_mode("___") %>%
  set_engine("___") %>% 
  fit(still_customer ~ ., data = ___) %>%
  predict(new_data = ___, type = "___") %>% 
  bind_cols(___)

# Calculate AUC
roc_auc(___, 
        truth = ___, 
        estimate = ___)
编辑并运行代码