帮我改写代码,使四种故障类型能够分别计算准确率并显示在四张图中:accTrain = sum(trainYreal_out==trainYreal)/length(trainYreal); figure plot(trainYreal,'o','linewidth',1.5,'markersize',7) hold on plot(trainYreal_out,'+','linewidth',1.5) title(['DBN训练集诊断准确率:' num2str(accTrain*100) '%'],'fontsize',12) xlabel('样本','fontsize',12) ylabel('类型','fontsize',12) legend('实际类型','诊断类型') grid on set(gca,'ytick',1:6,'yticklabel',names) set(gca,'linewidth',1.5) % 对测试数据进行测试 [testErr,testY_out] = nntest(nn, testXn, testY);
时间: 2023-07-16 13:13:09 浏览: 54
以下是修改后的代码。新增了一个for循环,分别计算四种故障类型的准确率,并在四张图中显示。
```
% 计算每种故障类型的准确率
accTrain = zeros(1,4);
for i = 1:4
idx = trainYreal == i;
accTrain(i) = sum(trainYreal_out(idx) == trainYreal(idx)) / sum(idx);
end
% 绘制四种类型的准确率图
figure
for i = 1:4
subplot(2,2,i)
idx = trainYreal == i;
plot(find(idx), trainYreal(idx), 'o', 'linewidth', 1.5, 'markersize', 7)
hold on
plot(find(idx), trainYreal_out(idx), '+', 'linewidth', 1.5)
title([names{i} '故障类型训练集诊断准确率:' num2str(accTrain(i)*100) '%'],'fontsize',12)
xlabel('样本','fontsize',12)
ylabel('类型','fontsize',12)
legend('实际类型','诊断类型')
grid on
set(gca,'ytick',1:6,'yticklabel',names)
set(gca,'linewidth',1.5)
end
% 对测试数据进行测试
[testErr,testY_out] = nntest(nn, testXn, testY);
```
相关推荐
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)