关于lable.shape的问题

[PTXXXXXZ]

想问lable的shape可以设置吗，是一直都是[128,14541]吗，如果想要其成为[128, ]该如何操作呢

[PTXXXXXZ]

head_label = torch.zeros(len(data), self.num_ent)
tail_label = torch.zeros(len(data), self.num_ent)
请问label设置成[128，14541]的话，KBAT的解码器不是convKB吗，最后如何eval的？

[Modberge]

在filter protocol情况下，label用来标记当前三元组中所有正确的头实体或尾实体。

目前NeuralKG支持link predication task，详细的eval过程请参照eval_task模块下的link_prediction.py

此外您也可以通过Translating Embeddings for Modeling Multi-relational Data论文了解模型是如何进行链接预测的。

希望这可以帮到您。