out的label问题

[1120161807]

trafficstars

您好！感谢您的代码！在运行过程中出现了一些问题想向您请教。 在main_mil.py中

1. max_ins_id = torch.max(out[:, label], 0)[1]这句有点不太理解，看了您的其他回答，"out是一个二维的tensor, 维度假设是B53, 表示一个bag内所有句子的预测概率."，比如一个bag中有两个句子，那么维度就是253，out[0]代表的就是第一个句子在每一种关系上的对应概率，对吗？
2. 如果是的话，"out[:, label]取出label那一列，然后再取最大值"，这个"label那一列"在哪呢？out中不是只有相应关系的概率值吗？

[ShomyLiu]

你好！

1. 是这样的， 在 max_ins_id = torch.max(out[:, label], 0)[1] out 表示这个bag内，每个句子的所有关系的概率值。
2. label 就是训练的标签。 比如label是3的话， 就在out中取出第三列，然后找到bag内所有句子在这一列上的最大的那个句子。

[zelin-x]

你好！

1. 是这样的， 在 max_ins_id = torch.max(out[:, label], 0)[1] out 表示这个bag内，每个句子的所有关系的概率值。
2. label 就是训练的标签。 比如label是3的话， 就在out中取出第三列，然后找到bag内所有句子在这一列上的最大的那个句子。

您好，请问这一步是在包中选择在包的标签上获得最大得分的instance，这样可以获取到每个包中一个instance,使用这些筛选得到的instance进行分类器的训练，是这样吗？