开始使用免费开始使用

构建一个 Logistic Regression 模型

您已经使用航班数据构建过一个决策树模型。现在,您将基于相同的数据创建一个 Logistic Regression 模型。

目标是预测某次航班是否可能延误至少 15 分钟(标签为 1),或不会(标签为 0)。

尽管您可以使用多种预测变量,此处暂时只使用 mondepartduration 三列。它们是数值特征,可以直接用于 Logistic Regression 模型。若要加入类别型特征,您还需要做一些前处理。请继续关注后续内容!

数据已被划分为训练集和测试集,分别为 flights_trainflights_test

本练习是课程的一部分

使用 PySpark 进行机器学习

查看课程

练习说明

  • 导入用于创建 Logistic Regression 分类器的类。
  • 创建分类器对象,并在训练数据上进行训练。
  • 对测试数据进行预测,并创建混淆矩阵。

交互式实操练习

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

# Import the logistic regression class
from pyspark.ml.____ import ____

# Create a classifier object and train on training data
logistic = ____().____(____)

# Create predictions for the testing data and show confusion matrix
prediction = ____.____(____)
prediction.groupBy(____, ____).____().show()
编辑并运行代码