-
Loss function for multi-label classificationData/Machine learning 2021. 5. 3. 20:01
Binary Cross Entropy is used for multi-label classification, and it's involved with a sigmoid function. (Note that cross-entropy is used for multi-class classification that's involved with a softmax function) (reference)
- Example of a simple neural network model for multi-label classification in Keras (link)
- Size of an output layer must be a number of labels. - [PyTorch web] BCEWithLogitLoss (link)
'Data > Machine learning' 카테고리의 다른 글
Evidence Lower Bound (ELBO) (0) 2021.08.24 Metrics for Multi-label classification (0) 2021.05.03 PyTorch Example of LSTM (0) 2021.04.21 - Example of a simple neural network model for multi-label classification in Keras (link)