분석시각화 대회 코드 공유 게시물은
내용 확인 후
좋아요(투표) 가능합니다.
2023 인하 인공지능 챌린지
baseline NCF 관련 질문.
ncf baseline으로 csv파일을 생성해본 결과 유저 아이디가 매칭이 안되어 있는데 그대로 제출해도 되는건가요?
혹시 recommend 함수의 하단부를 다음 코드로 교체 후에도, 같은 문제가 발생하는지 확인 부탁드립니다.
def recommend():
....
submit = pd.DataFrame(columns=["user_id", "item_id"])
user_list, item_list = [], []
for user, rec in tqdm(pred.items()):
user = parser.user_decoder[user]
for item in rec[0]:
item = parser.item_decoder[item]
user_list.append(int(user))
item_list.append(int(item))
submit["user_id"] = user_list
submit["item_id"] = item_list
return submit
안녕하세요 데이콘입니다.
위의 KDD님의 안내와 같이, NCF 베이스라인 코드에서 추론 부분의 recommend 함수 중
for item in rec: 부분을 for item in rec[0]: 로 수정하시면 올바른 결과를 얻을 수 있습니다.
해당 내용은 NCF 베이스라인 코드에 수정하여 반영하였습니다.
감사합니다.
데이콘(주) | 대표 김국진 | 699-81-01021
통신판매업 신고번호: 제 2021-서울영등포-1704호
직업정보제공사업 신고번호: J1204020250004
서울특별시 영등포구 은행로 3 익스콘벤처타워 901호
이메일 dacon@dacon.io |
전화번호: 070-4102-0545
Copyright ⓒ DACON Inc. All rights reserved
baseline ncf를 수행하시면, train.csv에서 rating 컬럼을 제외한 모습처럼 결과가 나오는 것이 정상입니다.
혹시 코드에서 별도의 수정은 없었을까요?