Neural Network Regression Problem with multiple Outputs

10 次查看(过去 30 天)
Hello everyone, I am now having a 1000 images, which have 10 continous attributes. My goal is to predict these 10 attributes values when I feed it a new image, after training.
So I think it is like a regression problem and I expect there will be 10 regression output layers in respond to these 10 attributes. How can i achieve this with Matlab? I think trainNetwork doesnt work. It is only applicable for non-multi-output lgraph.
Is that right direction? If not, can you point me?
I dont really know what parameters and state are.
I would be gladful if you leave a comment.

回答(1 个)

Srivardhan Gadila
Srivardhan Gadila 2020-3-14
The last two layers of your network architecture must be a fullyConnectedLayer with outputSize 10 followed by regressionLayer
layers = [ ...
fullyConnectedLayer(10)
regressionLayer];

类别

Help CenterFile Exchange 中查找有关 Image Data Workflows 的更多信息

标签

Community Treasure Hunt

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

Start Hunting!

Translated by