elm_train.m

来自「基于BP模型的神经网络模型」· M 代码 · 共 30 行

%~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
%
%  ELM_train(network,data) - train MNN model with ELM (standard mode) .
%  
%  Parameters: network -  neural network with matrix networks
%              data - training data sample
%
%  Author: Povilas Daniu餴s, [e-mail]
%  http://ai.hacker.lt - lithuanian site about Artificial Intelligence.
%
%  ----------------------------------------------------------------------


function f=ELM_train(network,data)

Y = data.target';

% calculate hidden layer output matrix
for i=1:length(Y)
    X(i,1) = 1;
    for j=2:network.regressors        
       X(i,j) = tanh( network.left(j).w * data.training(i).mat * network.right(j).w + network.bias(j) );
    end        
end    


% Batch mode
network.weights = inv(X'*X)*X'*Y;
f = network;