Побудуйте Decision Tree
Тепер, коли ви розділили дані про перельоти на тренувальну та тестову вибірки, можна використати тренувальну вибірку, щоб навчити модель Decision Tree.
Дані доступні як flights_train і flights_test.
ПРИМІТКА: Навчання моделі триватиме кілька секунд… будь ласка, зачекайте!
Ця вправа є частиною курсу
Machine Learning з PySpark
Інструкції до вправи
- Імпортуйте клас для створення класифікатора Decision Tree.
- Створіть об'єкт класифікатора та навчіть його на тренувальних даних.
- Зробіть прогнози для тестових даних і перегляньте результати.
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
# Import the Decision Tree Classifier class
from pyspark.ml.____ import ____
# Create a classifier object and fit to the training data
tree = ____()
tree_model = tree.____(____)
# Create predictions for the testing data and take a look at the predictions
prediction = tree_model.____(____)
prediction.select('label', 'prediction', 'probability').show(5, False)