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.

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

Z 2023-8-30

编辑：Z 2023-8-30

Hello Gagan,

This command doesn't work.

For example, if I do

net=googlenet;

net.layers{end}.size = 1;

I get: Unrecognized method, property, or field 'layers' for class 'DAGNetwork'.

请先登录，再进行评论。

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

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

回答（1 个）

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

另请参阅

类别

标签

产品

版本

Community Treasure Hunt

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

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

回答（1 个）

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

另请参阅

类别

标签

产品

版本

Community Treasure Hunt

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

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