Crie uma árvore de decisão
Agora que você dividiu os dados dos voos em conjuntos de treinamento e teste, pode usar o conjunto de treinamento para ajustar um modelo de árvore de decisão.
Os dados estão disponíveis em flights_train e flights_test.
NOTA: Vai demorar alguns segundos para o modelo ser treinado… por favor, aguarde!
Este exercício faz parte do curso
Machine Learning com PySpark
Instruções do exercício
- Importa a classe pra criar um classificador de árvore de decisão.
- Crie um objeto classificador e ajuste-o aos dados de treinamento.
- Faça previsões para os dados de teste e veja como ficaram.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# 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)