Hi Daeyeon Koh
The error message you're encountering indicates that the output of your network does not match the expected format of your target labels YTrain1. The network is configured to output 5 channels (due to numClass being 5, which leads to a fullyConnectedLayer with 5 outputs), but your target labels YTrain1 are likely not in a one-hot encoded format, which is expected for a multi-class classification problem when using crossentropy as the loss function.
To resolve this issue, make sure that your target labels YTrain1 are in the correct format. Since numClass is 5, each target should be a one-hot encoded vector of size 5. Here's how you can adjust your code:
1.) One-Hot Encode YTrain1: Convert YTrain1 into a one-hot encoded format. Assuming YTrain1 contains class labels as integers ranging from 1 to numClass, you can use the following code to one-hot encode these labels:
YTrain1_oneHot = full(ind2vec(YTrain1', numClass))';
2.) Adjust the Datastore for One-Hot Encoded Labels:
dsYTrain1 = arrayDatastore(YTrain1_oneHot, 'OutputType', 'same');
3.) Update the Loss Function
lossFcnA = @(Y1, T1) crossentropy(Y1, T1, 'DataFormat', 'BC');
Here, Y1 is the output of your network, and T1 is the one-hot encoded truth from dsYTrain1. The 'DataFormat', 'BC' argument specifies that both inputs and targets are in the format [BatchSize, NumClasses], which should match the format of your one-hot encoded labels.
Please refer to the following documentation link to learn more about ind2vec function.
Hope it helps!
