开始使用免费开始使用

评估随机森林

在最后一个练习中,您将评估随机森林模型的交叉验证结果。

以下对象已创建:

  • cv —— 已在训练数据上拟合的交叉验证器
  • evaluator —— 一个 BinaryClassificationEvaluator 对象,以及
  • flights_test —— 测试数据。

本练习是课程的一部分

使用 PySpark 进行机器学习

查看课程

练习说明

  • 打印参数网格中所有模型的平均 AUC 指标列表。
  • 显示最佳模型的平均 AUC。它是列表中"最大"的 AUC。
  • 打印最佳模型的 maxDepthfeatureSubsetStrategy 参数说明。
  • 显示最佳模型在测试数据上的预测 AUC。

交互式实操练习

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

# Average AUC for each parameter combination in grid
print(cv.____)

# Average AUC for the best model
print(____(____))

# What's the optimal parameter value for maxDepth?
print(cv.____.explainParam('____'))
# What's the optimal parameter value for featureSubsetStrategy?
print(cv.____.____(____))

# AUC for best model on testing data
print(evaluator.____(____.____(____)))
编辑并运行代码