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

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

상금 : 400 만원
1,172명 마감

연습

대회안내 데이터 코드 공유 토크 리더보드

제출

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

귀여운나등장

2023.11.17 20:56 1,593 조회 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

코드