Проблема с Matlab trainNetwork. Как правильно форматировать наблюдения? - PullRequest
0 голосов
/ 18 апреля 2019

Я пытаюсь использовать глубокую нейронную сеть в Matlab.Однако я получаю сообщение об ошибке: Ошибка при использовании trainNetwork (строка 165). Число наблюдений в X и Y не совпадают.

Ниже приведена моя текущая реализация:

size(XX) % ans = 120000 36
size(YY) % ans = 120000 24


%% Train a Deep Neural Network
layers = [
    imageInputLayer([1 36]); % Input is an "Image" 1x36 floating point vector
    fullyConnectedLayer(200);
    reluLayer();
    fullyConnectedLayer(100);
    reluLayer();
    fullyConnectedLayer(50);
    reluLayer();
    fullyConnectedLayer(24);
    regressionLayer();
];

if ~exist(checkpoint_directory_name,'dir')
    mkdir(checkpoint_directory_name)
end

options = trainingOptions('sgdm', ...
    'MaxEpochs', num_epochs, ... 
    'InitialLearnRate', 0.0001, ...
    'CheckpointPath', checkpoint_directory_name, ...
    'MiniBatchSize', batch_size ...
);

t = tic;
[net, info] = trainNetwork(XX, YY, layers, options);
fprintf('Trained in in %.02f seconds\n', toc(t));

Что можно сделать, чтобы решитьэта ошибка?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...