Construiește un arbore de decizie
Acum că ai împărțit datele despre zboruri în seturi de antrenament și de testare, poți folosi setul de antrenament pentru a antrena un model de tip arbore de decizie.
Datele sunt disponibile ca flights_train și flights_test.
NOTĂ: Modelul va avea nevoie de câteva secunde pentru a fi antrenat… ai puțină răbdare!
Acest exercițiu face parte din cursul
Machine Learning cu PySpark
Instrucțiuni pentru exercițiu
- Importă clasa folosită pentru a crea un clasificator de tip arbore de decizie.
- Creează un obiect clasificator și antrenează-l pe datele de antrenament.
- Generează predicții pentru datele de testare și aruncă o privire asupra rezultatelor.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# 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)