CNN: How to create label vector acceptable by trainNetwork

Question

Parichada Trairat 2021-8-15

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/898772-cnn-how-to-create-label-vector-acceptable-by-trainnetwork

回答： Anshika Chaurasia 2021-9-17

I'm trying to create cnn for load forecasting. I try to create lebel vector for training but it didn't work.

I have 3 types of data with 1920 samples fo each for training set (1920x3 double) and 3 types of data with 768 samples fo each for testing set (768x3 double). I reshaped train data to 4D array.

this is my code for this work.

%% Reshaped input
Train_dataset = reshape(Train_data, [1 1920 1 3]);
%  Creat the labels
label = (1:3,:);
Train_Labels = categorical(label); 
%% Layers
layers = [ ...
    imageInputLayer([1 1920 1])
    convolution2dLayer(1,20)
    reluLayer
    maxPooling2dLayer(1,'Stride',1)
    fullyConnectedLayer(1)
    softmaxLayer
    classificationLayer];
%% Train Options 
train_options = trainingOptions('sgdm');
%% Train network
net = trainNetwork(Train_dataset, Train_Labels, layers, train_options);
%% Test network
Pred = classify(net, Test_data)'