더북(TheBook)

마지막으로 정액 요금을 받는 운행을 어떻게 예측하는지 확인해 보자. 앞서 확인했듯이 JFK 공항을 오가는 택시는 운행 거리와 관계없이 52달러로 정해진 요금과 통행비를 받는다.

predict_random(df_prescaled, X_test, model)
Trip Details: Saturday, 23:00hrs
Actual fare: $52.00
Predicted fare: $53.55
RMSE: $1.55
116_1

▲ 그림 3-22 테스트 4: 공항 운행

이번 결과도 좋다. 신경망은 이 운행이 JFK 공항에서 출발했고 요금이 대략 52달러였다는 사실을 잘 학습한 것 같다. 특징 공학 과정에서 JFK 공항과 승차 및 하차 위치 사이의 거리를 계산해 새로운 특징 변수로 추가한 덕분이다. 신경망은 이 특징 변수를 단서로 JFK 공항을 오가는 요금이 대략 52달러라는 사실을 학습할 수 있다. 특징 공학이 이렇게 중요하다.

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.