제1회 국민대학교 AI빅데이터 분석 경진대회

알고리즘 | 정형 | 추천시스템 | Recall

  • moneyIcon Prize : 400 만원
  • 1,170명 마감

 

[AI스몰데이터]public 2th/ private 6th/(EASE+LightGCN+NGCF+WMF)

2023.11.17 20:56 1,544 Views language

cornac에 있는 모델들을 앙상블 했습니다.
BiVAECF, EASE, LightGCN, NGCF,WMF 모델들을 가지고 각각
BiVAECF, EASE, WMF 3개
BiVAECF, EASE, LightGCN, NGCF,WMF  5개 
그리고 제출한  BiVAECF 모델을 제외한 4개를 앙상블 했을 때 가장 잘 나왔습니다.

앙상블시 recruitment_seq 값들이 제법 겹쳐서 prediction 합의 값을 내림차순 순으로 동점인 것을 비교했습니다.
처음 recruitment_seq 개수만 뽑았을 때 보다 성능향상이 실제로 더 좋았습니다.

PDF
Code