MATLAB深度学习LSTM 标签分类classificationLayer层,YTrain的设置出错

问题:

使用matlab深度学习工具箱进行标签分类,设置XTrain为输入是n×1的元胞数组,每个元胞数据有6个特征值,YTrain输出是n×1的元胞数组,内容为标签'0'或‘1’。

LSTM层和options设置如下:

layers = [ ...
    sequenceInputLayer(inputsize)
    bilstmLayer(numHiddenUnits,'OutputMode','last')
    fullyConnectedLayer(numClasses)
    softmaxLayer
    classificationLayer];

options = trainingOptions('adam', ...
    'MaxEpochs',10, ...
    'GradientThreshold',2, ...
    'InitialLearnRate',0.005, ...
    'LearnRateSchedule','piecewise', ...
    'LearnRateDropPeriod',100, ...
    'LearnRateDropFactor',0.2, ...
    'Verbose',0, ...
    'Plots','training-progress',...
    'SequenceLength','longest', ...
    'MiniBatchSize',3);

利用深度学习工具进行lstm模型训练时,运行下方代码一直报这个错误。

net = trainNetwork(XTrain,YTrain,layers,options);

报错:

无效的训练数据。响应必须为分类响应的向量,或者为分类响应序列的元胞数组。

解决方案:

看了好多才发现自己有点呆,LSTM使用classificationLayer分类层时,YTrain需要使用标签元胞categoryCell,所以我的YTrain输入使用元胞数组、矩阵都是不行的,需要将ytrain转化为标签类型。使用categorical函数将ytrain改成标签类型。

dataTrain_output = categorical(dataTrain_output);

再重新训练就可以啦,开心~ 

  • 11
    点赞
  • 31
    收藏
    觉得还不错? 一键收藏
  • 13
    评论
评论 13
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值