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

1 次查看（过去 30 天）
Afiq Amirudin
Afiq Amirudin 2021-6-16
回答： 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 个）

Vineet Joshi
Vineet Joshi 2021-7-19
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.
  1. 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.
  2. 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
  3. 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

类别

在 Help Center 和 File Exchange 中查找有关 Deep Learning Toolbox 的更多信息

标签

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by