利用BCELoss解决Multi-label问题

class torch.nn.BCELoss(weight=None, size_average=True)[source] 计算 target 与 output 之间的二进制交叉熵。 l o s s ( o , t ) = − 1 n ∑ i ( t [ i ] l o g ( o [ i ] ) + ( 1 − t [ i ] ) l o g ( 1 − o [ i ] ) ) loss(o,
相关文章
相关标签/搜索