set labels for classifier deep learning toolbox
5 次查看(过去 30 天)
显示 更早的评论
Where do i set the labels for supervised training?
[XTrain,YTrain] = digitTrain4DArrayData;
idx = randperm(size(XTrain,4),1000);
XValidation = XTrain(:,:,:,idx);
XTrain(:,:,:,idx) = [];
YValidation = YTrain(idx);
YTrain(idx) = [];
layers = [
fullyConnectedLayer(4096,"Name","fc6","BiasInitializer","ones","WeightsInitializer","ones")
reluLayer("Name","relu6")
dropoutLayer(0.5,"Name","dropout6")
fullyConnectedLayer(4096,"Name","fc7")
reluLayer("Name","relu7")
dropoutLayer(0.5,"Name","dropout7")
fullyConnectedLayer(4096,"Name","fc8")
softmaxLayer("Name","softmax")
classificationLayer("Name","classoutput")];
plot(layerGraph(layers));
net = trainNetwork(XTrain,YTrain,layers,options);
0 个评论
回答(1 个)
Ankriti Sachan
2020-5-13
By setting the labels for supervised training, I am assuming that you want to ask how to train the dataset with the labeled data.
I guess you are already following this documentation - https://www.mathworks.com/help/deeplearning/ref/trainnetwork.html
Check the section here - https://www.mathworks.com/help/deeplearning/ref/trainnetwork.html#d120e82918
For the function,
net = trainNetwork(X,Y,layers,options), it is mentioned that
X = Training data
Y = Labels of the data that you are training the model with
layers = Neural network layers
option = Training options.
So I guess ‘Y’ is the variable that you are looking for.
1 个评论
Wiktor
2024-3-20
Hi! If I have augmentedImageDatastore created from imageDatadtore with defined Labels to every element, can I skip "Y" in trainNetwork function?
另请参阅
类别
在 Help Center 和 File Exchange 中查找有关 Image Data Workflows 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!