how do I set my fully connected layer to be (3 or 2) classification output?

Question

Afiq Amirudin 2021-6-16

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/857940-how-do-i-set-my-fully-connected-layer-to-be-3-or-2-classification-output

回答： Vineet Joshi 2021-7-19

Im working with pretrained network.

Currently, I have 3 age group (17-20, 21-40, 41-60) and another one is (female , male). My question is how to change the fully connected layer for this type of classification.

Currently I had 6 classes, 3 age group for each gender. Therefore, my classification output is 6 classes instead. Is this the correct way to classify?

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

Afiq Amirudin 2021-6-16

How to achieve something like this?

and how do my dataset structure should be?

请先登录，再进行评论。

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

Answer 1

Vineet Joshi 2021-7-19

1
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/857940-how-do-i-set-my-fully-connected-layer-to-be-3-or-2-classification-output#answer_749588

Hi

It is not possible to have one layer with arbitary number of neurons (3 or 2) . Keeping this in mind, there are three ways to approach your problem.

Create 6 labels (3 * 2) and train the network for classification on this problem. Since the number of instances are equal, you should get decent result but ensure to not let the model overfit.
You can have common layers upto a point and then split the network into two halves, one with output of 2 and other with output of 3. You can refer to the following example. Assemble Multiple-Output Network for Prediction
You can also have two seperate networks for both the predictions, but this is just the brute force way of doing the above.

Hope this was of some help.

Thanks