2023 인하 인공지능 챌린지

baseline NCF 관련 질문.

2023.07.09 13:02 1,405 조회
submit_ncf_03.csv

ncf baseline으로 csv파일을 생성해본 결과 유저 아이디가 매칭이 안되어 있는데 그대로 제출해도 되는건가요?

로그인이 필요합니다
0 / 1000
KDD
2023.07.10 14:44

baseline ncf를 수행하시면, train.csv에서 rating 컬럼을 제외한 모습처럼 결과가 나오는 것이 정상입니다.
혹시 코드에서 별도의 수정은 없었을까요?

KDD
2023.07.10 15:18

혹시 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

DACON.GM
2023.07.10 16:03

안녕하세요 데이콘입니다.
위의 KDD님의 안내와 같이, NCF 베이스라인 코드에서 추론 부분의 recommend 함수 중
for item in rec: 부분을 for item in rec[0]: 로 수정하시면 올바른 결과를 얻을 수 있습니다.
해당 내용은 NCF 베이스라인 코드에 수정하여 반영하였습니다.
감사합니다.