machine-learning - 自我错误中的索引超出范围 - PyTorch

Question

我正在构建一个推荐引擎，在底层使用 Pytorch（深度神经网络）。现在，我有一个很好的模型，我正在尝试生成一个表格，其中包含用户和内容喜欢能力（亲和力得分）之间的预测。

我的历史交互输入表

user_id | content_id
1       | 23
1       | 20
2       | 34

然后，我将这些列转换为张量：

users=input['user_id].to_list()
content=input['content_id].to_list()

#convert to tensors
user_id_tensor = torch.tensor(user_id)
content_id_tensor = torch.tensor(content_id)

#prediction using historical data input on the trained model object
prediction = model.forward(user_id_tensor,content_id_tensor)

Error: Index out of range in self

知道问题是什么吗？

标签： machine-learningpytorchtensor