MIO Lab2. E2

e02.m

clc; clear;
%% train
data_train = load('Lab2_training.txt');
x_train = data_train(:, 1:4)';
y_train = data_train(:, 5:7)';
%% test
data_test = load('Lab2_testing.txt');
x_test = data_test(:, 1:4)';
y_test = data_test(:, 5:7)';

%% learning
n = [30 20];

net = feedforwardnet(n);
net.divideFcn = 'dividetrain';
%net.layers{0}.transferFcn = 'logsig';
% net.layers{1}.transferFcn = 'tansig';
net.layers{2}.transferFcn = 'tansig';
net.trainParam.epochs = 100;
disp(net);

net = train(net, x_train, y_train);

r_test = net(x_test);
disp(r_test);
e = mse(net, r_test, y_test);

disp(['mse = ', num2str(e)]);