1. Обучение
  2. /
  3. Курса
  4. /
  5. 使用 PySpark 进行机器学习

Connected

упражнение

构建 Decision Tree

现在您已经将航班数据拆分为训练集和测试集,可以用训练集来拟合一个 Decision Tree 模型。

数据分别存为 flights_train 和 flights_test。

注意:模型训练需要几秒钟……请耐心等待!

Инструкции

100 XP
  • 导入用于创建 Decision Tree 分类器的类。
  • 创建一个分类器对象,并将其拟合到训练数据上。
  • 对测试数据进行预测,并查看预测结果。