1. Learn
    2. /
  2. Courses
    3. /
  3. PySpark로 하는 Machine Learning
Connected

Exercise

Decision Tree 구축하기

이제 항공편 데이터를 학습용과 테스트용으로 분할했으니, 학습용 데이터로 Decision Tree 모델을 학습시켜 보세요.

데이터는 flights_train과 flights_test로 제공됩니다.

NOTE: 모델 학습에 몇 초 정도 걸릴 수 있어요. 잠시만 기다려 주세요!

Instructions

100 XP
  • Decision Tree 분류기를 생성하는 클래스를 임포트하세요.
  • 분류기 객체를 만들고 학습용 데이터에 맞춰 학습시키세요.
  • 테스트용 데이터에 대해 예측을 수행하고 예측 결과를 확인하세요.