시작하기무료로 시작하기

모델 블렌딩

이제 블렌딩 기법으로 모델 앙상블을 만들어 보겠습니다.

목표는 New York City Taxi 대회의 데이터를 사용해 서로 다른 2개의 모델을 학습하고, 테스트 데이터에 대한 예측을 만든 다음 간단한 산술 평균으로 블렌딩하는 것입니다.

traintest DataFrame은 작업 공간에 이미 준비되어 있습니다. features는 학습에 사용할 열 이름 목록이며 작업 공간에 제공되어 있습니다. 타깃 변수 이름은 "fare_amount"입니다.

이 연습은 강의의 일부입니다

Python으로 Kaggle 대회 공략하기

강의 보기

연습 안내

  • features 목록을 사용하고 타깃 변수로 "fare_amount" 열을 지정해 train 데이터에 Gradient Boosting 모델을 학습하세요.
  • 같은 방식으로 Random Forest 모델을 학습하세요.
  • Gradient Boosting과 Random Forest 모델을 사용해 test 데이터에 대한 예측을 만드세요.
  • 두 모델의 예측값 평균을 구하세요.

실습형 인터랙티브 연습

이 예제를 이 샘플 코드를 완성하여 풀어보세요.

from sklearn.ensemble import GradientBoostingRegressor, RandomForestRegressor

# Train a Gradient Boosting model
gb = GradientBoostingRegressor().____(____[features], ____.fare_amount)

# Train a Random Forest model
rf = RandomForestRegressor().____(____[features], ____.fare_amount)

# Make predictions on the test data
test['gb_pred'] = ____.____(test[features])
test['rf_pred'] = ____.____(test[features])

# Find mean of model predictions
test['blend'] = (____[____] + ____[____]) / 2
print(test[['gb_pred', 'rf_pred', 'blend']].head(3))
코드 편집 및 실행