clear all; clc; % 载入数据 data = xlsread('Copy_of_数据集.xlsx'); input = data((1:120), 2:6)'; output = data((1:120), 7:9)'; % 划分训练集和测试集 input_train = input(:, 1:80); output_train = output(:, 1:80); input_test = input(:, 81:100); output_test = output(:, 81:100); % 归一化 [input_train_n, input_ps] = mapminmax(input_train, -1, 1); [output_train_n, output_ps] = mapminmax(output_train, -1, 1); % 建立模型 input_size = size(input_train_n, 1); hidden_size = 10; output_size = size(output_train_n, 1); net = newff(input_train_n, output_train_n, hidden_size, {'tansig','purelin'}, 'trainlm'); net.trainParam.epochs = 15000; net.trainParam.lr = 0.01; net.trainParam.goal = 0.0001; % 训练模型 [net, tr] = train(net, input_train_n, output_train_n); % 测试模型 input_test_n = mapminmax('apply', input_test, input_ps); output_test_n = mapminmax('apply', output_test, output_ps); output_pred_n = sim(net, input_test_n); %% 反归一化 output_test_pred = mapminmax('reverse', output_pred_n, output_ps); output_test_pred = round(output_test_pred); % 四舍五入取整 % 使用测试集评估网络性能 pos_pred = sim(net, input_test_n); % 预测位置 ori_pred = sim(net, input_test_n); % 预测姿态 pos_error = pos_pred - output_test(1,:); % 位置误差 ori_error = ori_pred - output_test(1,:); % 姿态误差 mse_pos = mean(pos_error.^2); % 位置均方误差 mse_ori = mean(ori_error.^2); % 姿态均方误差 % 使用附加测试集评估网络性能 % additional_test_data = [theta([6, 12, 18], :), actual_poses([6, 12, 18], :)]; additional_test_data = input(101:120,:)'; additional_test_data_n = mapminmax('apply', additional_test_data, input_ps); pos_pred = sim(net, additional_test_data_n); % 预测位置 ori_pred = sim(net, additional_test_data_n); % 预测姿态 pos_error = pos_pred - output(1,:); % 位置误差 ori_error = ori_pred - output(1,:); % 姿态误差 mse_pos_additional = mean(pos_error.^2); % 位置均方误差 mse_ori_additional = mean(ori_error.^2); % 姿态均方误差 % 调整维度为 2 x 10 % 绘制预测结果和真实结果的对比图 figure; plot(output_test(1,:), 'bo-'); hold on; plot(output_test_pred(1,:)', 'r*-'); % 注意转置 legend('真实结果', '预测结果'); xlabel('样本编号'); ylabel('输出值'); title('预测结果和真实结果');显示additional_test_data = input(101:120,:); 位置 1 处的索引超出数组边界(不能超出 5)。
时间: 2024-01-04 20:04:35 浏览: 58
arima_test.zip_ARIMA代码_arima.test_arima预测_stationary test_平稳性检验
5星 · 资源好评率100%
这段代码中出现了一个错误,提示“位置 1 处的索引超出数组边界(不能超出 5)。”,意味着在使用“additional_test_data = input(101:120,:)”这行代码时,访问了数组边界之外的元素。根据错误提示,数组的大小为5,因此访问索引大于5的元素会导致错误。可能的原因是数据集的大小与代码中的索引范围不一致,需要检查数据集的大小并修正代码中的索引范围。
阅读全文