分类：nn.BCELoss

2020
09-23

细数nn.BCELoss与nn.CrossEntropyLoss的区别

以前我浏览博客的时候记得别人说过，BCELoss与CrossEntropyLoss都是用于分类问题。可以知道，BCELoss是BinaryCrossEntropyLoss的缩写，BCELossCrossEntropyLoss的一个特例，只用于二分类问题，而CrossEntropyLoss可以用于二分类，也可以用于多分类。不过我重新查阅了一下资料，发现同样是处理二分类问题，BCELoss与CrossEntropyLoss是不同的。下面我详细讲一下哪里不同。1、使用nn.BCELoss需要在该层前面加上Sigmoid函数。公式... 继续阅读 >