how to train LSTM with single input and two outputs?

조회 수: 14 (최근 30일)
Muhammad Faisal Khalid
Muhammad Faisal Khalid 2021년 11월 24일
답변: Prateek Rai 2022년 2월 22일
hello everyone,
I have question regarding the training of LSTM network. I want to train my network with 1 input and 2 outputs.
Network architecture is as:
layers = [ ...
sequenceInputLayer(numFeatures,'Normalization', 'zscore')
lstmLayer(numHiddenUnits,'OutputMode','sequence')
lstmLayer(numHiddenUnits,'OutputMode','sequence')
lstmLayer(numHiddenUnits2,'OutputMode','sequence')
lstmLayer(numHiddenUnits2,'OutputMode','sequence')
fullyConnectedLayer(numResponses)
regressionLayer];
with numFeatures=1 and numResponses=2.
Do i have to make custom regression layer for 2 output as i read that for multiple input and single output, custom regression layer is needed to train the network but there is no information for multiple out.
anybody can help me in this regard.
Thanks.

답변 (1개)

Prateek Rai
Prateek Rai 2022년 2월 22일
To my understanding, you want to train LSTM with two outputs.
You can refer to MATLAB Answer on LSTM Example for Multi input and Multi outputs to get more idea.

카테고리

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