Построение дерева решений
Теперь, когда данные о рейсах разделены на обучающую и тестовую выборки, можно обучить модель дерева решений на обучающей выборке.
Данные доступны в виде flights_train и flights_test.
ПРИМЕЧАНИЕ: Обучение модели займёт несколько секунд — пожалуйста, подождите!
Это упражнение является частью курса
Машинное обучение с PySpark
Инструкции к упражнению
- Импортируйте класс для создания классификатора на основе дерева решений.
- Создайте объект классификатора и обучите его на тренировочных данных.
- Получите предсказания для тестовой выборки и изучите результаты.
Интерактивное практическое упражнение
Попробуйте выполнить это упражнение, дополнив этот пример кода.
# 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)