1. Nauka
    2. /
  2. Kursy
    3. /
  3. Uczenie maszynowe z PySpark
Connected

ćwiczenie

Budowanie drzewa decyzyjnego

Po podziale danych o lotach na zbiór treningowy i testowy możesz użyć zbioru treningowego do wytrenowania modelu drzewa decyzyjnego.

Dane są dostępne jako flights_train i flights_test.

UWAGA: Trenowanie modelu może potrwać kilka sekund – uzbrój się w cierpliwość!

Instrukcje

100 XP
  • Zaimportuj klasę służącą do tworzenia klasyfikatora drzewa decyzyjnego.
  • Utwórz obiekt klasyfikatora i dopasuj go do danych treningowych.
  • Wygeneruj przewidywania dla danych testowych i sprawdź wyniki.