항공편 소요 시간 모델: 거리만 사용
이번 연습에서는 회귀 모델을 만들어 항공편 소요 시간(duration 열)을 예측해 보겠습니다.
우선은 모델을 단순하게 유지하여, 예측 변수로 항공편 거리(km 열)만 포함하겠습니다.
데이터는 flights에 있습니다. 처음 몇 개의 레코드는 터미널에 표시되어 있습니다. 이 데이터는 이미 학습용과 테스트용으로 분할되어 있으며, flights_train과 flights_test로 사용할 수 있습니다.
이 연습은 강의의 일부입니다
PySpark로 하는 Machine Learning
연습 안내
- 선형 회귀 객체를 생성하세요. 레이블 열 이름을 지정하고, 학습 데이터에 적합하세요.
- 테스트 데이터에 대해 예측을 수행하세요.
- 회귀 평가자 객체를 생성하고, 이를 사용해 테스트 데이터의 RMSE를 평가하세요.
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
from pyspark.ml.regression import LinearRegression
from pyspark.ml.evaluation import RegressionEvaluator
# Create a regression object and train on training data
regression = ____(____).____(____)
# Create predictions for the testing data and take a look at the predictions
predictions = ____.____(____)
predictions.select('duration', 'prediction').show(5, False)
# Calculate the RMSE
____(____).____(predictions)