Bygg ett beslutsträd
Nu när du har delat upp flygdata i tränings- och testmängder kan du använda träningsdata för att anpassa en beslutsträdsmodell.
Data finns tillgängliga som flights_train och flights_test.
OBS: Det tar några sekunder för modellen att tränas – ha tålamod!
Den här övningen är en del av kursen
Maskininlärning med PySpark
Övningsinstruktioner
- Importera klassen för att skapa en beslutsträdsklassificerare.
- Skapa ett klassificerare-objekt och anpassa det till träningsdata.
- Gör förutsägelser för testdata och granska resultaten.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# 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)