CCF图书推荐系统NCF代码运行错误

问题遇到的现象和发生背景

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)
运行结果及报错内容

img

有截图吗发过来看一下