0
$\begingroup$

Say I use a (convolutional) neural network to classify a multi-label problem. For example, the problem is classifying different stages of skin cancer from 0 (healthy) to 4 (severe). The five stages are represented in the following manner.

0 => [1, 0, 0, 0, 0]
1 => [1, 1, 0, 0, 0]
2 => [1, 1, 1, 0, 0]
3 => [1, 1, 1, 1, 0]
4 => [1, 1, 1, 1, 1]

In the last layer of my neural network, I have 5 output neurons that use a sigmoid activation function.

After I've trained my neural network, I feed it a test image and it gives me the following prediction.

[0.6, 0.2, 0.7, 0.1, 0.1]

If any class output greater than 0.5 activates a class, I get the following prediction.

[1, 0, 1, 0, 0]

My question is, should this correspond to a prediction of 2, 1 or 0?

$\endgroup$

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Browse other questions tagged or ask your own question.