CCF图书推荐系统NCF的代码运行报错啊
def chunks(l, n):
for i in range(0, len(l), n):
yield l[i, i + n]
f = open('D:/MachineLearning/book-recommender/submission.csv', 'w', encoding='utf-8')
for user in user_for_test:
# 将用户已经交互过的物品排除
user_visited_items = traindataset.user_book_map[user]
items_for_predict = list(set(range(traindataset.book_nums)) - set(user_visited_items))
results = []
user = torch.Tensor([user]).to(device)
for batch in chunks(items_for_predict, 64):
batch = torch.Tensor(batch).unsqueeze(0).to(device)
result = model(user, batch).view(-1).detach().cpu()
results.append(result)
有截图吗发过来看一下