For binary classification using a neural network, I want to enforce one output node instead of two output nodes

Question

Z 2022-9-21

1
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1809515-for-binary-classification-using-a-neural-network-i-want-to-enforce-one-output-node-instead-of-two-o

编辑： Z 2023-8-30

I have a dataset consisting of a positive class and a negative class which I want to classifiy using a neural network (e.g. cancerous tumor or benign). I want to set the classification layer to a single node, either this class or that class. During training, MATLAB detects that the data contains two labels, and insists that I put two nodes at the output. How can I enforce one node at the output?

1 个评论
显示 -1更早的评论隐藏 -1更早的评论

Z 2022-9-22

I learned that I needed to use a sigmoid instead of a softmax before the output layer. Now the issue is that I cannot use the built-in classification layer, and need to create a custom layer which uses the binary cross-entropy loss.

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Gagan Agarwal 2023-8-30

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1809515-for-binary-classification-using-a-neural-network-i-want-to-enforce-one-output-node-instead-of-two-o#answer_1296891

在 MATLAB Online 中打开

Hi Z

In MATLAB, the default assumption is that your dataset contains multiple classes, and as a result, the output layer is automatically configured accordingly with the desired no of nodes to accommodate the corresponding classes.

However, if you specifically want to ensure that the output layer has only one node, you can achieve this by explicitly setting the size of the output layer to 1 using the following command:

net.layers{end}.size = 1;

Here, the variable 'net' represents your neural network.