HD현대 AI Challenge

알고리즘 | 채용 | 정형 | 조선해양 | 회귀 | MAE

  • moneyIcon 상금 : 2,000만 원
  • 1,510명 마감

 

[Private 5th] AutoML(autogluon) + Bertime_predictions

공동작성자

stroke
2023.11.18 14:23 1,140 조회 language

배가 항구에 도착시간 BERT_TIME :: ATA - CI_HOUR = BERTIME
자세한 과정 : https://github.com/ColdTbrew/HD_Hyundai_AI_Challenge.git

1. start training (train_auto06.ipynb)

train_auto06.ipynb 파일을 실행시키면 autogluon 모델을 학습시키고 test파일에 대해 predict하여 회귀모델이 예측한 값을 auto06.csv파일로 저장합니다. 
2. train셋의 Bert_time을 이용하여 test셋의 가능 CI_HOUR 도출 (make_use_bertTime_and_get_CI_HOURS.ipynb)

train셋에서 Bert_time 열을 만들고 필요한 열만 추출하여 small_train 데이터프레임을 만듭니다.
병렬처리를 이용해 train셋의 기후컬럼 6열과 test셋의 기후컬럼 6열이 일치하면 test셋에서 (train의 Bert_time 값 - test셋의 ATA 값)을 적용해 CI_HOUR를 계산하여 리스트에 저장합니다. 이 때, 리스트에 저장되는 가능한 CI_HOUR 값은 여러개가 될 수 있고 없을 경우에 -1을 저장합니다. 또한, 값이 음수가 되는 것은 저장하지 않습니다.
완료되면 use_bertTime_and_get_CI_HOURS.csv 파일로 저장합니다.

코드