월간 데이콘 항공편 지연 예측 AI 경진대회

알고리즘 | 정형 | 분류 | 준지도학습 | 항공 | LogLoss

  • moneyIcon Prize : 인증서
  • 1,543명 마감

 

TabNet을 이용한 예측 코드

2023.05.01 21:34 2,789 Views language

TabNet은 정형데이터를 다루는 데 특화된 딥러닝 모델 중 하나입니다. TabNet은 기본적으로 트리 기반 모델인 Decision Tree와 유사한 구조를 가지고 있지만, Decision Tree와는 달리 그래디언트 부스팅 대신에 어텐션 메커니즘을 사용합니다. 이러한 특징으로 인해, TabNet은 다양한 종류의 특성(Features) 간에 상호작용을 효과적으로 학습할 수 있으며, 특성 선택과 피쳐 중요도를 자동으로 학습합니다.
아래 코드는 베이스라인 코드에 TabNet을 적용한 것 입니다.

Code