Error using trainNetwork (line 184) Number of observations in X and Y disagree.

Please help me, I'm getting this error when I tried to train my network. Thank you. I tried the code from this Reference

Error using trainNetwork (line 184)

Number of observations in X and Y disagree.

%18. train the network

miniBatchSize = 128;
validationFrequency = floor(numel(YTrain)/miniBatchSize);
options = trainingOptions('adam', ...
    'InitialLearnRate',3e-4, ...
    'MaxEpochs',30, ...
    'MiniBatchSize',miniBatchSize, ...
    'Shuffle','every-epoch', ...
    'Plots','training-progress', ...
    'Verbose',false, ...
    'ValidationData',{XValidation,YValidation}, ...
    'ValidationFrequency',validationFrequency, ...
    'LearnRateSchedule','piecewise', ...
    'LearnRateDropFactor',0.1, ...
    'LearnRateDropPeriod',20);

whos XTrain YTrain XValidation YValidation Xbkg

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

[KSSV]

What are the dimensions of XTRain and YTrain?

[Sharena Natasha Nor Hisham]

@KSSV Here are the dimensions

Name Size Bytes Class Attributes

XTrain 98x50x1x170 3332000 single

XValidation 98x50x1x30 588000 single

Xbkg 98x50x1x200 3920000 single

YTrain 2856x1 4534 categorical

YValidation 1050x1 2728 categorical

[KSSV]

You have messed up with the imensions of XTrain and YTrain. XTrain has 170 inputs, so likewise for each 170 XTrain you are supposed to have 170 YTrain. But you have YTrain as 2856x1. There is no one-to-one correspondence between input an output.

[Sharena Natasha Nor Hisham]

oh how can I fix the dimensions?

[KSSV]

It depends on what your problem is. Regression or classification? What data you have? what you are expecting?